Как хранить 2d матричные значения в переменной?

Question

import numpy as np
import pandas as df
from numpy import asarray
from numpy import save
files=np.load('arr.npy',allow_pickle=True)

#print(files)
data=df.DataFrame(files)

type(data)
rr=data.shape[0]

for i in range(0,rr):

      res=data[0][i]

после запуска переменная res содержит последний элемент

, но я хочу, чтобы все значения

, поэтому скажите мне, как сохранить все значения матрицы 2d в python ??

переменная данных - это фрейм данных

, он содержит 9339 строк и 2 столбца

, но я хочу, чтобы 1-й столбец это матрица 32x32

как хранить значения переменной res

Answer 1

Обратите внимание, что res = data[0][i] инициализирует новую переменную res на первой итерации l oop (когда i равно 0), но затем продолжает переназначать ее значение на значение в следующей строке ( оставаясь в столбце 0).

Я не совсем уверен, что вы хотите, но звучит так, как будто вы просто хотите первый столбец в отдельной переменной? Вот как можно получить первый столбец, как pandas ряд и / или простой список, с меньшим примером (9 строк и 2 столбца)

import pandas as pd
random_data = np.random.rand(9,2)
data_df = pd.DataFrame(random_data)
print(data_df)

# this gets the first column as a pandas series. Change index from 0 to get another column.
print('\nfirst column:')
first_col = data_df[data_df.columns[0]]
print(first_col)

# if you want a plain list instead of a series
print('\nfirst column as list:')
print(first_col.tolist())

Вывод:

          0         1
0  0.218237  0.323922
1  0.806697  0.371456
2  0.526571  0.993491
3  0.403947  0.299652
4  0.753333  0.542269
5  0.365885  0.534462
6  0.404583  0.514687
7  0.298897  0.637910
8  0.453891  0.234333

first column:
0    0.218237
1    0.806697
2    0.526571
3    0.403947
4    0.753333
5    0.365885
6    0.404583
7    0.298897
8    0.453891
Name: 0, dtype: float64

first column as list:
[0.21823726509923325, 0.8066974875381492, 0.526571422644495, 0.40394686954663594, 0.7533330239460391, 0.36588470364914194, 0.4045827678891364, 0.2988970490642284, 0.45389073978613426]