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]
浙公网安备 33010602011771号