如何在numpy中的列上执行公式

时间:2019-09-30 00:33:01

标签: python numpy-ndarray

我有一个2d ndarray,我需要将每列转换为一个列表。

我在2d数组下贴了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 :(得分:0)

只需尝试

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