博客

如何从字符串列表中检查Pandas列是否有值

  • Post category:Python

要从一个字符串列表中检查Pandas列是否有值，可以使用 Pandas 的 isin() 方法。

具体过程如下：

  1. 创建一个字符串列表，例如：list = ['value1', 'value2', 'value3']
  2. 使用 Pandas 读取数据，例如：df = pd.read_csv('data.csv')
  3. 使用 Pandas 的 isin() 方法，检查列是否有值，例如：df['column'].isin(list)
  4. 将结果保存为一个新的 DataFrame，例如：new_df = df[df['column'].isin(list)]

以下是一个具体的例子：

import pandas as pd

# 创建一个字符串列表
my_list = ['New York', 'London', 'Tokyo']

# 读取数据
data = {'city': ['New York', 'London', 'Paris', 'Tokyo', 'Sydney'],
        'temperature': [25, 18, 20, 15, 30]}
df = pd.DataFrame(data)

# 检查列是否有值
new_df = df[df['city'].isin(my_list)]

# 打印结果
print(new_df)

输出结果为：

        city  temperature
0   New York           25
1     London           18
3      Tokyo           15

可以看到，只有包含在字符串列表中的城市数据被保留在新的 DataFrame 中。

Tags: