如何修复 Python 散点图中的 "list index out of range" 错误？

Question

你能看看我的代码并告诉我我收到 'list index out of range' 错误的原因吗？

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 列表，该迭代器迭代 colors 列表的 len。

我已经尝试使用示例数据，我将其发布在下方。希望对你有帮助！！

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])

你会得到这样的输出：

如何修复 Python 散点图中的 "list index out of range" 错误？

How can I fix "list index out of range" error in my scatter plot on Python?

python

scatter-plot