如何将ndarray转换为数组的元组?

时间:2020-02-26 09:53:36

标签: python-3.x tuples numpy-ndarray

例如,如果我有一个ndarray(mp),例如:

`mp =
 [[2.3980985828756736 6861 -1 6861]
 [2.4120089384661587 6862 -1 6862]
 [2.4235102880216637 6863 -1 6863]
 ...
 [1.0564918793423272 2418 2418 -1]
 [1.0562595999589373 2419 2419 -1]
 [1.0543968001562891 2420 2420 -1]]`

我想要mp的前两列,并希望将它们像这样转换

the mp is 
 `(array([2.12734638, 2.13031872, 2.12896207, ..., 4.24341706, 3.71451699,
       3.24234188]), array([ 77795.,  77796.,  77797., ..., 111854.,  60951.,  60952.]))`

我该如何提前感谢...

1 个答案:

答案 0 :(得分:1)

您可以了解有关numpy索引here

的信息

例如,您可以使用:

mp = np.array(
 [[2.3980985828756736, 6861, -1, 6861],
 [2.4120089384661587, 6862, -1, 6862],
 [2.4235102880216637, 6863, -1, 6863],
 [1.0564918793423272, 2418, 2418, -1],
 [1.0562595999589373, 2419, 2419, -1],
 [1.0543968001562891, 2420, 2420, -1]])

print((mp[:, 0], mp[:, 1]))

输出:

(array([2.39809858, 2.41200894, 2.42351029, 1.05649188, 1.0562596 ,
       1.0543968 ]), array([6861., 6862., 6863., 2418., 2419., 2420.]))