使用query
方法检查列的元素是否包含特定字符串可以通过以下步骤实现:
import pandas as pd
data = {'Column1': ['apple', 'banana', 'orange'],
'Column2': ['cat', 'dog', 'elephant'],
'Column3': ['red apple', 'yellow banana', 'orange juice']}
df = pd.DataFrame(data)
query
方法检查列的元素是否包含特定字符串,例如检查Column1
列是否包含字符串apple
:result = df.query('Column1.str.contains("apple")', engine='python')
在上述代码中,str.contains
函数用于检查字符串是否包含特定子字符串,engine='python'
参数用于指定使用Python解释器执行查询。
print(result)
完整的代码示例:
import pandas as pd
data = {'Column1': ['apple', 'banana', 'orange'],
'Column2': ['cat', 'dog', 'elephant'],
'Column3': ['red apple', 'yellow banana', 'orange juice']}
df = pd.DataFrame(data)
result = df.query('Column1.str.contains("apple")', engine='python')
print(result)
输出结果:
Column1 Column2 Column3
0 apple cat red apple
在这个例子中,我们创建了一个包含三列数据的DataFrame对象。然后,我们使用query
方法检查Column1
列的元素是否包含字符串apple
,并将结果存储在result
变量中。最后,我们打印出结果,只包含包含字符串apple
的行。
腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云