Как мне выполнить формулу для столбца в NumPy - PullRequest
0 голосов
/ 30 сентября 2019

У меня есть 2d ndarray, который мне нужно преобразовать каждый столбец в список.

Я разместил 2 фрагмента кода ниже одного, который возвращает мне только первую ячейку, а второй - первое вхождение в массив 2d, мой массив выглядит так:

 ```
 [[ 2.29  0.09]
 [ 7.12  1.2 ]
 [ 6.53  0.  ]
 [ 8.79  2.12]
 [ 2.51  0.  ]]
 ```

Я пробовал это: tmp_lst = np.array(cne_arr[0,0]).tolist()

и это np.array(cne_arr[0]).tolist() Первый код сверху получает меня первый столбец первого ряда: 2.29

, а второй получает меня весь первый ряд: 2.29 0.09

i need 2 lists one that is the first column
  (2.29 7.12 6.53 8.79 2.51 )

and one that is the second column
  (0.09 1.2 0.  2.12 0.) 

Я уверен, что здесь должен быть легкий путь, но я новичок в python.

1 Ответ

0 голосов
/ 30 сентября 2019

Просто попробуйте

tmp = # the array 
col_1 = tmp[:, 0].tolist()
col_2 = tmp[:, 1].tolist()
...