Python中列表的分类选择

Question

我在表格中有数字和颜色的列表：

示例：

1.9   green  black  
2.3   red    black
3.3   green  black
8.2   red    black
0.1   green  white
10.2  green  white

我正在尝试编写一种选择结构，以对不同列表中不同类型的颜色进行排序。

这就是我得到的：

`import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
df = pd.read_csv('hw03_problem2.csv' , skiprows=1)
df = pd.DataFrame(columns=['num','gb','bw'])

#df=df.astype(float)
f, ((ax1, ax2), (ax3, ax4)) = plt.subplots(2, 2, sharex='col', sharey='row')
print(df)
df[df.gb == 'green'].num.plot.hist(ax = ax1)
df[df.gb == 'red'].num.plot.hist(ax = ax2)
df[df.bw == 'black'].num.plot.hist(ax = ax3)
df[df.bw == 'white'].num.plot.hist(ax = ax4)


plt.show()
plt.savefig('subplot_for_q2')
~                                       

〜
〜
〜
〜

Answer 1

尚不清楚您要做什么。您为什么不尝试使用df = pd.read_csv（'hw03_problem2.csv'，names = ['number'，'color1'，'color2']）读取数据，这将使处理起来更加容易。

然后可以使用

绘制特定的子集

df[df.color1 == 'black'].number.plot.hist(ax=ax1)

（如果已经设置了轴）。假设这就是您要绘制的直方图的数字。