例如,如果我有一个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.]))`
我该如何提前感谢...
答案 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.]))