arr = np.array([[1,2,3],[3,4,5],[4,5,6]])
print (arr)

[[1 2 3]
 [3 4 5]
[4 5 6]]

  

行:

单行

row_1 = arr[0]

多行

row_1_2 = arr[0:2]
print(row_1_2)
最后一位取不到

条件

# 选择第1列>3的行
condition = arr[:,0] >3
select_row = arr[condition]
print(select_row)

  

列:

单列

# 选取第1列
col_1 = arr[:,0]

  

多列

最后一位取不到
col_1_2 = arr[:,0:2]

  

条件

# 选取第2行,大于3的列
condititon = arr[1,:]>3
arr[condititon]

  

posted on 2023-11-20 18:14  黑逍逍  阅读(13)  评论(0)    收藏  举报