数据框列到列表列表[重复]答案

【问题标题】：Dataframe column to list of list [duplicate]数据框列到列表列表[重复]
【发布时间】：2021-03-27 03:12:58
【问题描述】：

我想将数据框列转换为列表列表，如下所示：

df = pd.read_csv('xyz.csv')
print(df.head())
dataset = df.passenger_count.values.astype('float32')
print(dataset[:10])

输出：

     month   passenger_count
0  1949-01        112
1  1949-02        118
2  1949-03        132
3  1949-04        129
4  1949-05        121

[112. 118. 132. 129. 121. 135. 148. 148. 136. 119.]

我希望输出是这样的：

[[112.] [118.] [132.] [129.] [121.] [135.] [148.] [148.] [136.] [119.]]

【问题讨论】：

使用df[['passenger_count']].to_numpy().tolist()

标签： python pandas dataframe

【解决方案1】：

你只需要重塑你的 numpy 数组，它就会如你所愿：

dataset.reshape((-1,1)).tolist()
>[[112.], 
  [118.],
  [132.],
  [129.],
  [121.],
  [135.],
  [148.],
  [148.],
  [136.],
  [119.]]

或在 cmets 中建议 @cs96 ：

df[['passenger_count']].to_numpy().tolist()

会产生同样的结果...

注意：如果您想保留为 numpy 数组，请避免使用 tolist()...

【讨论】：