你可以通过 Pandas 的 filter()
函数结合 like
参数来筛选出列名中包含特定字符串(例如 ESP
)的列。以下是示例代码:
import pandas as pd# 创建一个示例 DataFrame
data = {'ESP_A': [1, 2, 3],'B_ESP': [4, 5, 6],'C': [7, 8, 9],'ESP_Device': [10, 11, 12]
}
df = pd.DataFrame(data)# 使用 filter() 找出所有名字包含 'ESP' 的列
esp_columns = df.filter(like='ESP')
print(esp_columns)
解释
filter(like='ESP')
:like
参数用于匹配列名中包含指定字符串的列。这里我们找的是包含'ESP'
的列。- 返回的
esp_columns
是一个新的 DataFrame,包含所有列名中有'ESP'
的列。
输出
ESP_A B_ESP ESP_Device
0 1 4 10
1 2 5 11
2 3 6 12
这样你就可以轻松筛选出名字包含 'ESP'
的列了。