Filtering for 'contains value' in a dataset in Python

user_12741271 0
Viewed 604

In Python how do you filter a column by values that contain a particular value?

An example is a data set with a column called 'City' and the values could be 'Sydney', 'Greater Sydney', 'North Sydney' etc. If a use input 'Sydney' how can I ensure that all variations are included in the filtering?

#user inputs column
input1 = input()
country_city = input1.title()

#user inputs value
input2 = input()
country_city_value = input2.title()

#filtering step (current)
filtered = dataset[dataset[country_city] == country_city_value]
print(filtered)
pandas python
2 Answers

If you want to filter contents that contain the input word, apply the str.contains:

data = {'City': ['Sydney', 'Greater Sydney', 'North Sydney'],}

dataset = pd.DataFrame(data, columns = ['City'])

#user inputs column
input1 = input()              
country_city = input1.title()        # 'City'

#user inputs value
input2 = input()
country_city_value = input2.title()  # 'Sydney'

#filtering step (current)
filtered = dataset[dataset[country_city].str.contains(country_city_value)]

#              City
# 0          Sydney
# 1  Greater Sydney
# 2    North Sydney
print(filtered)
user_141488640

str.contains is good way to go, but if your input will be 'North Sydney' you will not receive Sydney as a result , only north sydney example:

df= pd.DataFrame({
    'A':['Sydney','North Sydney','Alaska']

})
print(df)
              A
0        Sydney
1  North Sydney
2        Alaska
input='North Sydney'
filtered = df[df.A.str.contains(input)]

print(filtered)
              A
1  North Sydney

So to improve that way use split() with str.contains()

input=input.split()
print(input)
['North', 'Sydney']

filtered = df[df.A.str.contains('%s'%[x for x in input])]

print(filtered)
              A
0        Sydney
1  North Sydney

So in this way , you are sure that all parts of your input will be takeen into account

user_85601670
Related