使用包含标签的 CSV 文件绘制图表

Question

我正在尝试创建一个从包含标签的CSV 文件中绘制图表的函数：

LVL,Highscore
2,6013
2,5934
1,132
3,9560
4,13406

这是我试过的代码sn-p：

from matplotlib import style
import numpy as np
import csv

def plot():
    style.use('ggplot')

    x, y = np.loadtxt(
        'highscore.csv',
         unpack=True,
         skiprows=1,
         usecols=1,
         delimiter=','
    )

plot()

这是我得到的回溯错误：

Traceback (most recent call last):
  File "C:/Users/User/PycharmProjects/FinalYearProject/anothertest.py", line 28, in <module>
    plot()
  File "C:/Users/User/PycharmProjects/FinalYearProject/anothertest.py", line 18, in plot
    x, y = np.loadtxt('highscore.csv', unpack=True,
  File "C:\Users\User\PycharmProjects\FinalYearProject\venv\lib\site-packages\numpy\lib\npyio.py", line 1148, in loadtxt
    for x in read_data(_loadtxt_chunksize):
  File "C:\Users\User\PycharmProjects\FinalYearProject\venv\lib\site-packages\numpy\lib\npyio.py", line 992, in read_data
    vals = [vals[j] for j in usecols]
  File "C:\Users\User\PycharmProjects\FinalYearProject\venv\lib\site-packages\numpy\lib\npyio.py", line 992, in <listcomp>
    vals = [vals[j] for j in usecols]
IndexError: list index out of range

我尝试使用参数usecols=1 和skiprows=1 来解决它，但它似乎仍然不起作用。

Answer 1

不确定我是否完全理解您的问题，但我认为您不需要 usecols。

>>> import numpy as np
>>> x, y = np.loadtxt('highscore.csv', unpack=True,  skiprows=1, delimiter=',')

>>> print(x)

[2., 2., 1., 3., 4.]

>>> print(y)

[ 6013.,  5934.,   132.,  9560., 13406.]

然后您可以继续绘制数据。我不确定您期望哪个图，但散点图似乎最适合这里

>>> import matplotlib.pyplot as plt
>>> from matplotlib import style

>>> plt.style.use('ggplot')

>>> plt.scatter(x, y)
>>> plt.show()

这将是什么情节：

这能回答你的问题吗？