在 python 中打印數據頭的最簡單方法？

Question

我沒有用 pandas 定義我的數組，我正在使用 numpy 來做它，我想知道是否有任何其他方法可以打印數據的前 5 行。 使用 pandas 我會這樣做：打印（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, :])