numpy数组 传入 int 和 传入 list[] 的区别

numpy数组 传入 int 和 传入 list[] 的区别

pip install bpython
$ bpython
bash 命令行运行，直接上代码

import numpy as np

>>> a = np.array([[1, 2, 3], [3, 4, 5], [111, 222, 333], [333, 0, 0]])
>>> print(a)
[[  1   2   3]
 [  3   4   5]
 [111 222 333]
 [333   0   0]]
# 传入 int
>>> print(a[1][0])
3
# 传入 list
>>> print(a[[1]][0])
[3 4 5]

# 可以看到，numpy 允许 传入 [] 来取对应行

# 取 第 [1,2,3] 行得到一个新的 numpy 数组
>>> print(a[[1,2,3]])
[[  3   4   5]
 [111 222 333]
 [333   0   0]]


>>> print(a[[1,2,3]][0])
[3 4 5]
>>> print(a[[1,2,3]][1])
[111 222 333]
>>> print(a[[1,2,3]][2])
[333   0   0]
>>> 
 
 


  

 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
  
10
  
11
  
12
  
13
  
14
  
15
  
16
  
17
  
18
  
19
  
20
  
21
  
22
  
23
  
24
  
25
  
26
  
27
  
28
  
29
  
30
  
31
  
32
  
33
  
34
 

文章来源: positive.blog.csdn.net，作者：墨理学AI，版权归原作者所有，如需转载，请联系作者。

原文链接：positive.blog.csdn.net/article/details/119776843