布爾索引（Boolean Indexing）是數據分析中一種強大且常用的技術，用于通過布爾值數組（即包含 True 和 False 的數組）來選擇數據子集。布爾索引可以用于 NumPy 數組、Pandas 數據框等數據結構。

布爾索引是一種非常有用的技術，可以通過條件篩選數據。它不僅簡化了代碼，還提高了數據操作的效率。無論是在 NumPy 還是 Pandas 中，布爾索引都能幫助你快速、簡潔地選擇和操作數據子集。

NumPy 中的布爾索引

在 NumPy 中，布爾索引可以用于選擇滿足特定條件的數組元素。例如：

import numpy as np# 創(chuàng)建一個 NumPy 數組
data = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])# 創(chuàng)建一個布爾數組，條件是元素大于 5
bool_array = data > 5# 使用布爾索引選擇元素
filtered_data = data[bool_array]print("Original data:", data)
print("Boolean array:", bool_array)
print("Filtered data:", filtered_data)

輸出：

Original data: [ 1  2  3  4  5  6  7  8  9 10]
Boolean array: [False False False False False  True  True  True  True  True]
Filtered data: [ 6  7  8  9 10]

Pandas 中的布爾索引

在 Pandas 中，布爾索引可以用于選擇滿足特定條件的行或列。例如：

import pandas as pd# 創(chuàng)建一個 Pandas 數據框
data = {'A': [1, 2, 3, 4, 5],'B': [10, 20, 30, 40, 50],'C': [100, 200, 300, 400, 500]}
df = pd.DataFrame(data)# 創(chuàng)建一個布爾數組，條件是列 'A' 的值大于 2
bool_series = df['A'] > 2# 使用布爾索引選擇行
filtered_df = df[bool_series]print("Original DataFrame:")
print(df)
print("\nBoolean Series:")
print(bool_series)
print("\nFiltered DataFrame:")
print(filtered_df)

輸出：

Original DataFrame:A   B    C
0  1  10  100
1  2  20  200
2  3  30  300
3  4  40  400
4  5  50  500Boolean Series:
0    False
1    False
2     True
3     True
4     True
Name: A, dtype: boolFiltered DataFrame:A   B    C
2  3  30  300
3  4  40  400
4  5  50  500