Категориальный отбор для списка в 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)

, если у вас уже настроены оси.Предполагается, что это числа, из которых вы делаете гистограмму.