Получить информацию из массива numpy

Question

У меня есть массив numpy размером 5 на 182, который выглядит примерно так:

matched_data = '/home/myname/data.csv'
matched_data = Table.read(matched_data, format="ascii")

ID_1 = np.array(matched_data['name'])
ID_1.astype(str)

ID_2 = np.array(matched_data['col1_1'])
ID_2.astype(str)

mag = np.array(matched_data['Magnitude'])
mag.astype(float)

semi_major_axis = np.array(matched_data['SMA'])
semi_major_axis.astype(float)

semi_minor_axis = np.array(matched_data['sma'])
semi_minor_axis.astype(float)

position_angle = np.array(matched_data['pos_ang'])
position_angle.astype(float)


match = (ID_1, ID_2, mag, (semi_major_axis/semi_minor_axis), position_angle)
match = np.array(match) 

Как бы я мог получать данные из этой строки построчно? Например, когда я набираю:

print(match[0])

, он выводит список идентификаторов из ID_1, где, как мне бы хотелось, ID_1, ID_2, величина, (semi_major_axis / semi_minor_axis) и position_angle первого объекта. Как бы я это сделал?

Answer 1

Вместо match = np.array(match) вы можете использовать match = np.stack(match, axis=1). Таким образом, различные количества (ID_1, ID_2, mag, ...) будут столбцами результирующего массива, и вы можете получить к ним доступ через match[index]. Если вы сделали np.stack(match, axis=0), то это похоже на то, что вы делали изначально, и данные будут результирующими массивами. Здесь все еще возможно получить к нему доступ, указав индекс столбца: match[:, 0].