0
import pandas as pd


data = {'name': ['HelloWorld', 'ByeWorld'],
        'physics': [22, 33],
        'chemistry': [44, 55]}

a = pd.DataFrame(data)

b = a.loc[a['name'] == 'Hello']

print(b)

This code would not return any rows, but I would like to achieve it would return the first row, because it includes "Hello". Is there any elegant solution to solve this with the loc command?

Henry Ecker
  • 34,399
  • 18
  • 41
  • 57
filtertips
  • 801
  • 3
  • 12
  • 25

1 Answers1

1

You want

b = a.loc[a['name'].str.contains('Hello')]

also if your only looking at the start of a string you can use

b = a.loc[a['name'].str.startswith('Hello')]

Not

b = a.loc[a['name'] == 'Hello']

this line of code will only return True for a row containing ONLY Hello

Kenan
  • 13,156
  • 8
  • 43
  • 50