python - 如何从数据框创建列表?

标签 python pandas

我有一个坐标为 x0,y0,x1,y1 的数据框。我正在尝试获取坐标列表

datafrme 就像 -

      x0     y0      x1      y1      x2      y2 ...
 1  179.0   77.0    186.0   93.0    165.0   91.0... 
 2  178.0   76.0    185.0   93.0    164.0   91.0...

期望的输出

 [  [ [179.0,77.0], [186.0,93.0], [165.0,91.0  ],...  ],
    [ [178.0,76.0 ],[185.0,93.0 ],[164.0,91.0 ] ,...  ]
 ]

我正在尝试获取按行的坐标列表

我已经尝试过了

df = pd.read_csv(path,header=None)
df.reset_index(drop=True, inplace=True)
data = df.values.tolist()

感谢您的所有见解!

最佳答案

如果性能很重要,请使用reshape:

a = len(df.columns)
b = len(df)

L = df.values.reshape(b, a).reshape(b, a // 2, 2).tolist()
print (L)
[[[179.0, 77.0], [186.0, 93.0], [165.0, 91.0]], 
 [[178.0, 76.0], [185.0, 93.0], [164.0, 91.0]]]

关于python - 如何从数据框创建列表?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52983061/

相关文章:

python - Pandas:将日期划分为 30 分钟间隔并计算平均值

python - 根据第二个数据帧设置数据帧的值

python - 将数据框列转换为数组

Python Pandas 按连续多列分组

java - 算法,从包含 n 个元素的列表中查找组合

python - 为什么断言没有大量使用?

python - 在嵌套 while 循环之间切换

python - Pandas groupby 和基于多列的转换

python - 通过聚合函数过滤 Pandas 数据框?

python - Pandas:从函数连续写入 csv