如何使用pandas python將行轉換爲列表？

Question

product,count,value1,value2,value3 
A,10,5,3,2 
B,8,2,2,4 

這是我的數據幀。我需要一個像下面的格式輸出：

product,count,values 
A,10,[5,3,2] 
B,8,[2,2,4] 

Answer 1

這裏有一種方法

In [27]: df['values'] = df[['value1', 'value2', 'value3']].values.tolist() 

In [28]: df 
Out[28]: 
    product count value1 value2 value3  values 
0  A  10  5  3  2 [5, 3, 2] 
1  B  8  2  2  4 [2, 2, 4] 

In [29]: df.drop(['value1', 'value2', 'value3'], axis=1) 
Out[29]: 
    product count  values 
0  A  10 [5, 3, 2] 
1  B  8 [2, 2, 4] 

詳情：

In [35]: df = pd.DataFrame([['A', 10, 5, 3, 2], ['B', 8, 2, 2, 4]], 
    ....:     columns=['product', 'count', 'value1', 'value2', 'value3']) 

In [36]: df 
Out[36]: 
    product count value1 value2 value3 
0  A  10  5  3  2 
1  B  8  2  2  4