выбор столбца из сводной таблицы панд - PullRequest
Новая
       8

выбор столбца из сводной таблицы панд

0 голосов
/ 07 сентября 2018

У меня есть приведенная ниже сводная таблица, которую я создал из кадра данных, используя следующий код: 

table = pd.pivot_table(df, values='count', index=['days'],columns['movements'], aggfunc=np.sum)

movements 0    1   2   3   4   5   6   7
days
0         2777 51  2                    
1         6279 200 7   3                
2         5609 110 32  4                
3         4109 118 101 8   3            
4         3034 129 109 6   2   2        
5         2288 131 131 9   2   1        
6         1918 139 109 13  1   1        
7         1442 109 153 13  10  1        
8         1085 76  111 13  7           1
9         845  81  86  8   8            
10        646  70  83  1   2   1   1

Как видно из сводной таблицы, в ней 8 столбцов от 0 до 7, и теперь я хочу построить некоторые конкретные столбцы вместо всех. Мне не удалось выбрать столбцы. Допустим, я хочу построить столбец 0 и столбец 2 против индекса. что я должен использовать для y, чтобы выбрать столбец 0 и столбец 2? 

plt.plot(x=table.index, y=??)

Я пробовал с y = table.value['0', '2'] и y=table['0','2'], но ничего не работает.

1 Ответ

0 голосов
/ 07 сентября 2018

Вы не можете выбрать ndarray для y, если вам нужны эти два значения столбца на одном графике, который вы можете использовать: 

plt.plot(table['0'])
plt.plot(table['2'])

Если имена столбцов являются целыми числами, то: 

plt.plot(table[0])
plt.plot(table[2])
...