如何在python中比较两列字符串?

时间:2018-10-08 14:35:05

标签: python pandas

我的CSV文件包含20列,我只需要获取与我的研究相关的那些地址的数据,因此我将包含所有地址的列与仅包含特定地址的列进行比较。

我收到“关键错误”,表明索引selected_city不存在:

import csv
import os
import pandas as pd
data_new = pd.read_csv('file1.csv', encoding= "ISO-8859–1")
print(data_new)
for i in rows:
    if str(data.loc['selected_city'] == data.loc['Charge_Point_City'])
print(data.Volume,data.Charge_Point_City)

1 个答案:

答案 0 :(得分:0)

考虑使用内置函数.isin()

例如:

s = pd.Series(['a','b','c', 'b','c','a','b'])

所以现在看起来像:

0    a
1    b
2    c
3    b
4    c
5    a
6    b

假设您只想保留s中较小系列的行:

smol = pd.Series(['a','b'])
s[s.isin(smol)]

输出:

0    a
1    b
3    b
5    a
6    b

对于您的特定用例,您可能需要

data = data[data['selected_city'].isin(data['Charge_Point_City'])]