Создать новый столбец на основе строки

Question

У меня есть фрейм данных, я хочу создать столбец на основе строки в column1_sport.

import pandas as pd

df = pd.read_csv('C:/Users/test/dataframe.csv', encoding  = 'iso-8859-1')

Данные содержат:

column1_sport
baseball
basketball
tennis
boxing
golf

Я хочу найти определенные строки («шарик» или «коробка») и создать новый столбец на основе того, содержит ли столбец это слово. Если датафрейм не содержит этого слова, добавьте «other». Смотри ниже.

column1_sport    column2_type
baseball         ball
basketball       ball
tennis           other 
boxing           box              
golf             other

Answer 1

Для этого простого случая вы можете создать собственный словарь и использовать его для map Серии df.column1_sport:

d = {'basketball':'ball', 'boxing':'box', 'baseball':'ball'}
df['column2_type'] = df.column1_sport.map(d).fillna('other') 

    column1_sport column2_type
0      baseball         ball
1    basketball         ball
2        tennis        other
3        boxing          box
4          golf        other

Answer 2

df["column2_type"] = df.column1_sport.apply(lambda x: "ball" if "ball" in x else ("box" if "box" in x else "Other"))
df

    column1_sport   column2_type
0        baseball           ball
1      basketball           ball
2          tennis          Other
3          boxing            box
4            golf          Other

Если у вас есть более сложные условия

def func(a):
    if "ball" in a.lower():
        return "ball"
    elif "box" in a.lower():
        return "box"
    else:
        return "Other"

df["column2_type"] = df.column1_sport.apply(lambda x: func(x))

Answer 3

Для нескольких условий я предлагаю np.select. Например:

values = ['ball', 'box']
conditions = list(map(df['column1_sport'].str.contains, values))

df['column2_type'] = np.select(conditions, values, 'other')

print(df)

#   column1_sport column2_type
# 0      baseball         ball
# 1    basketball         ball
# 2        tennis        other
# 3        boxing          box
# 4          golf        other

Answer 4

Вы можете использовать вложенный np.where

cond1 = df.column1_sport.str.contains('ball')
cond2 = df.column1_sport.str.contains('box')
df['column2_type'] = np.where(cond1, 'ball', np.where(cond2, 'box', 'other') )

    column1_sport   column2_type
0   baseball        ball
1   basketball      ball
2   tennis          other
3   boxing          box
4   golf            other