如何使用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