根据变量的类别计数绘制数据集

Question

我有一个包含 14 列的数据集（我只需要使用 4 列：旅行舱位、性别、年龄和票价），我将其拆分为火车和测试数据集。我需要从火车数据集中创建一个垂直条形图，用于按旅行等级（1、2 和 3 是等级）的乘客分布。我不允许使用 NumPy、Pandas、SciPy 和 SciKit-Learn。

我对 Python 很陌生，我知道如何绘制非常简单的图形，但是当涉及到更复杂的图形时，我会有点迷茫。

这是我的代码（我知道有很多错误）：

travelling_class = defaultdict(list)
for row in data:
    travelling_class[row[0]]

travelling_class = {key: len(val) for key, val in travelling_class.items()}

keys = travelling_class()
vals = [travelling_class[key] for key in keys]
ind  = range(min(travelling_class.keys()), max(travelling_class.keys()) + 1)
width = 0.6

plt.xticks([i + width/2 for i in ind], ind, ha='center')
plt.xlabel('Tracelling Class') 
plt.ylabel('Counts of Passengers')
plt.title('Number of Passengers per Travelling Class')
plt.ylim(0, 1000)
plt.bar(keys, vals, width)
plt.show()

---

import matplotlib.pyplot as plt

classes = travelling_class[1, 2, 3]

plt.hist(classes)
plt.show()

@TrakJohnson 这是问题的最初提问者 - 抱歉，我不小心以某种方式删除了我的个人资料，所以不得不创建一个新的。非常感谢你的帮助。问题是我的数据集是 1045 行，所以可能很难列出所有这些。以上看起来合理吗？

Answer 1

使用plt.hist，将绘制直方图(more info here)

例子：

import matplotlib.pyplot as plt

classes = [1, 2, 1, 1, 3, 3]

plt.hist(classes)
plt.show()

结果如下：