在 python 中打印数据头部的最简单方法？

Question

我没有用 pandas 定义我的数组，我正在使用 numpy 来做它，我想知道是否有任何其他方法来打印数据的前 5 行。使用 pandas 我会这样做： print(data.head()).

这是我定义数据的方式：

with open('B0_25.txt', 'r') as simulation_data:
simulation_data = [x.strip() for x in simulation_data if x.strip()]

data = [tuple(map(float, x.split())) for x in simulation_data[2:100]]  

x = [x[1] for x in data]
y = [x[2] for x in data]
z = [x[3] for x in data]
mx = [x[4] for x in data]
my = [x[5] for x in data]
mz = [x[6] for x in data]

mydata = np.array([x, y, z, mx, my, mz])

Answer 1

您需要对 mydata 进行转置，否则 x, y, z, mx, my, mz 是行而不是列。

mydata = np.array([x, y, z, mx, my, mz]).T
print(mydata[:5, :])

在 python 中打印数据头部的最简单方法？

Easiest way to print the head of a data in python?

python

pandas

head