Самый простой способ распечатать заголовок данных в 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, :])