python - 使用csv文件作为输入在python中绘制直方图

Question

我有一个包含两列的 csv 文件，其中第一列是水果名称，第二列是计数，我需要使用此 csv 作为以下代码的输入来绘制直方图。我如何让它成为可能。我只需要显示前 20 个条目，其中水果名称将是 x 轴，计数将是 100 行的整个 csv 文件中的 y 轴。

import matplotlib.pyplot as plt
import pandas as pd

data = pd.read_csv('data.csv', header = None ,quoting=2)

data.hist(bins=10)
plt.xlim([0,100])
plt.ylim([50,500])
plt.title("Data")
plt.xlabel("fruits")
plt.ylabel("Frequency")
plt.show()

我编辑了上面的程序来绘制条形图 -

import matplotlib.pyplot as plt
import pandas as pd

data = pd.read_csv('data.csv', sep=',',header=None)
data.values
print data
plt.bar(data[:,0], data[:,1], color='g')
plt.ylabel('Frequency')
plt.xlabel('Words')
plt.title('Title')

plt.show()

但这给了我一个错误“Unhashable Type”。任何人都可以提供帮助。

Answer 1

你可以使用pandas的内置绘图，虽然你需要指定第一列是索引，

import matplotlib.pyplot as plt
import pandas as pd

data = pd.read_csv('data.csv', sep=',',header=None, index_col =0)

data.plot(kind='bar')
plt.ylabel('Frequency')
plt.xlabel('Words')
plt.title('Title')

plt.show()

如果您需要使用matplotlib，使用data.to_dict() 将数组转换为字典并提取数据到numpy 数组或其他东西可能会更容易.