我没有用pandas定义我的数组,我正在使用numpy做到这一点,我想知道是否还有其他方法可以打印数据的前5行。使用熊猫,这就是我的方法: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])
您需要转置mydata,否则x,y,z,mx,my,mz是行而不是列。
mydata = np.array([x, y, z, mx, my, mz]).T
print(mydata[:5, :])