在数组中创建行的元组的最简单方法是什么?

时间:2016-04-11 17:31:24

标签: python numpy

说我有一个X by 2 np.ndarray。

1 2
3 4
3 6

从行中创建元组的最简单方法是什么

  (1 ,2)
  (3, 4)
  (3, 6)

除了

L = [ tuple(A[i,:]) for i in range(X)]

L = array(L)

是否有内置功能来执行此操作?

1 个答案:

答案 0 :(得分:3)

由于numpy数组按行迭代,所以只需使用普通的旧列表解析即可:

>>> a = np.arange(1, 7).reshape((3,2))
>>> a
array([[1, 2],
       [3, 4],
       [5, 6]])
>>> [tuple(row) for row in a]
[(1, 2), (3, 4), (5, 6)]

你也可以map

>>> map(tuple, a)
[(1, 2), (3, 4), (5, 6)]