python将列表转换成集合_python – 将dataFrame转换为list

我有一个pandas数据帧,我转换为numpy数组,如下所示：

df.values

它给出了以下输出：

array([[2],

[0],

[1],

...,

[0],

[1],

[0]], dtype=int64)

但是我想获得如下列表：

[0, 2, 3]

知道怎么做吗？

解决方法:

也许您可以使用iloc或loc来选择列,然后使用tolist：

print df

a

0 2

1 0

2 1

3 0

4 1

5 0

print df.values

[[2]

[0]

[1]

[0]

[1]

[0]]

print df.iloc[:, 0].tolist()

[2, 0, 1, 0, 1, 0]

或者可能：

print df.values.tolist()

[[2L], [0L], [1L], [0L], [1L], [0L]]

print df.iloc[:, 0].values.tolist()

[2L, 0L, 1L, 0L, 1L, 0L]

print df.loc[:, 'a'].tolist()

[2, 0, 1, 0, 1, 0]

print df['a'].tolist()

[2, 0, 1, 0, 1, 0]

但也许你需要flatten：

print df.values.flatten()

[2 0 1 0 1 0]

print df.iloc[:, 0].values.flatten()

[2 0 1 0 1 0]

标签：python,dataframe,pandas

来源： https://codeday.me/bug/20190516/1116049.html