import numpy as np

# 創(chuàng)建一個(gè)3行4列的數(shù)組（元素0-11）
a = np.arange(12).reshape(3, 4)  
print("原數(shù)組a：")
print(a)
# Output:
# [[ 0  1  2  3]
#  [ 4  5  6  7]
#  [ 8  9 10 11]]

# 創(chuàng)建與a同形的布爾數(shù)組：判斷a中元素是否大于4
b = a > 4  
print("\n布爾數(shù)組b（a > 4的結(jié)果）：")
print(b)
# Output:
# [[False False False False]
#  [False  True  True  True]
#  [ True  True  True  True]]

# 用布爾數(shù)組b索引a，提取所有True位置的元素（返回一維數(shù)組）
selected_elements = a[b]  
print("\n篩選出的元素：")
print(selected_elements)
# Output: [ 5  6  7  8  9 10 11]

# 繼續(xù)使用上面的數(shù)組a和布爾數(shù)組b
a[b] = 0  # 將a中所有>4的元素賦值為0
print("修改后的數(shù)組a：")
print(a)
# Output:
# [[0 1 2 3]
#  [4 0 0 0]
#  [0 0 0 0]]

import numpy as np

a = np.arange(12).reshape(3, 4)  # 3行4列，行索引0-2，列索引0-3
print("原數(shù)組a：")
print(a)
# Output:
# [[ 0  1  2  3]
#  [ 4  5  6  7]
#  [ 8  9 10 11]]

# 行選擇的布爾數(shù)組：長度=行數(shù)（3），選擇第2、3行（索引1、2）
b1 = np.array([False, True, True])  
# 列選擇的布爾數(shù)組：長度=列數(shù)（4），選擇第1、3列（索引0、2）
b2 = np.array([True, False, True, False])  

# 選擇行（第一個(gè)維度）：等價(jià)于a[b1, :]
print("\n選擇行（b1為[False, True, True]）：")
print(a[b1])  
# Output:
# [[ 4  5  6  7]
#  [ 8  9 10 11]]

print("\n選擇列（b2為[True, False, True, False]）：")
print(a[:, b2])
# Output:
# [[ 0  2]
#  [ 4  6]
#  [ 8 10]]

print("\n同時(shí)選擇行和列（b1和b2結(jié)合）：")
print(a[b1, b2])
# Output: [4 10]
# 解釋：b1選擇行1、2，b2選擇列0、2 → 取(1,0)和(2,2)位置的元素：4和10