如何修復 Python 散點圖中的“列表索引超出范圍”錯誤？

Question

你能看看我的代碼並告訴我為什么我收到“列表索引超出范圍”錯誤的原因嗎？

fig, ax = plt.subplots(figsize=(10,5))
view_count = dodo_data['view_count']
like_count = dodo_data['like_count']
colors = ['r', 'k', 'b', 'g']

ax.legend()
ax.set_xlabel('View Count')
ax.set_ylabel('Like Count', rotation=0, labelpad=55)

for i in range(len(view_count)):
    plt.plot(view_count[i], like_count[i], 'o', color=colors[i])

Answer 1

問題出在colors[i]上。 您已使用i遍歷view_count列的長度。 而是嘗試使用單獨的迭代器來索引colors列表，該迭代器len colors列表的長度。

我已經嘗試使用我在下面發布的示例數據。 希望能幫助到你！！

from pandas import DataFrame

Data = {'view_count':  [1,2,3,4,5],
        'like_count': [6,7,8,9,10],
        }

dodo_data = DataFrame (Data, columns = ['view_count','like_count'])

print(dodo_data)

import matplotlib.pyplot as plt
import random

fig, ax = plt.subplots(figsize=(10,5))
view_count = dodo_data['view_count']
like_count = dodo_data['like_count']
colors = ['r', 'k', 'b', 'g']

ax.legend()
ax.set_xlabel('View Count')
ax.set_ylabel('Like Count', rotation=0, labelpad=55)

for i in range(len(view_count)):
    plt.plot(view_count[i], like_count[i], 'o', color=colors[random.sample(range(len(colors)),1)[0]])

Answer 2

你可以這樣做：

for i in range(len(view_count)):
    plt.plot(view_count[i], like_count[i], 'o', color=colors[i%4])

你會得到這樣的輸出：

1 6 r
2 7 k
3 8 b
4 9 g
5 10 r
6 2 k
7 3 b
8 4 g
9 5 r
34 4 k