首頁猿問我需要將列列表中的每個元素轉換為...

我需要將列列表中的每個元素轉換為 python pandas 中的新列

Python

慕的地6264312 2023-09-19 17:04:53

我有一個 Python 數據框，如下所示： Name Hobbies0 Paul ["Watch_NBA", "Play_PS4"]1 Jeff ["Play_hockey", "Read", "Play_PS4"]2 Kyle ["Sleep", "Watch_NBA"]我需要將列表中的每個元素轉換到新列中，如果它出現在原始列表中，則分配值 0 或 1。結果顯示如下： Name Watch_NBA Play_PS4 Play_hockey Read Sleep0 Paul 1 1 0 0 01 Jeff 0 1 1 1 02 Kyle 1 0 0 0 1有人知道我怎么能做到這一點。請記住，我將在專欄中使用很多愛好，因此它顯示出一點自動化而不是硬編碼。謝謝?。?！

查看完整描述

4 回答

阿波羅的戰車

TA貢獻1862條經驗獲得超6個贊

get_dummies()很好，但sklearn's?MultiLabelBinarizer有更好的性能：

from sklearn.preprocessing import MultiLabelBinarizer

mlb = MultiLabelBinarizer()

a = mlb.fit_transform(df["Hobbies"])

df_expanded = pd.DataFrame(a, columns=mlb.classes_, index=df.index)

# merge them using the following:

df_merged = df.merge(df_expanded, left_index=True, right_index=True)

print(df_merged)

index? ?Name? ? Hobbies? ? ? ? ? ? ? ? ? ? ? ? ?Play_PS4? ? Play_hockey Read? ? Sleep? ?Watch_NBA

0? ? ? ?Paul? ? [Watch_NBA, Play_PS4]? ? ? ? ? ?1? ? ? ? ? ?0? ? ? ? ? ?0? ? ? ?0? ? ? ?1

1? ? ? ?Jeff? ? [Play_hockey, Read, Play_PS4]? ?1? ? ? ? ? ?1? ? ? ? ? ?1? ? ? ?0? ? ? ?0

2? ? ? ?Kyle? ? [Sleep, Watch_NBA]? ? ? ? ? ? ? 0? ? ? ? ? ?0? ? ? ? ? ?0? ? ? ?1? ? ? ?1

反對回復 2023-09-19

慕姐4208626

TA貢獻1852條經驗獲得超7個贊

In [86]: df

Out[86]:

Name Hobbies

0 Paul [NBA, PS4]

1 Jeff [Hockey, Read, PS4]

2 Kyle [Sleep, NBA]

In [87]: df['dummy'] = 1

In [88]: df.explode("Hobbies").pivot(index='Name', columns='Hobbies', values='dummy').fillna(value=0)

Out[88]:

Hobbies Hockey NBA PS4 Read Sleep

Name

Jeff 1.0 0.0 1.0 1.0 0.0

Kyle 0.0 1.0 0.0 0.0 1.0

Paul 0.0 1.0 1.0 0.0 0.0

反對回復 2023-09-19

子衿沉夜

TA貢獻1828條經驗獲得超3個贊

你想要get_dummies()方法。

對于你的例子：

names = df.Name

df = pd.get_dummies(df.Hobbies.apply(pd.Series).stack()).sum(level=0)

df.insert(0, 'Name', names)

#output:

? ?Name? Play_PS4? Play_hockey? Read? Sleep? Watch_NBA

0? Paul? ? ? ? ?1? ? ? ? ? ? 0? ? ?0? ? ? 0? ? ? ? ? 1

1? Jeff? ? ? ? ?1? ? ? ? ? ? 1? ? ?1? ? ? 0? ? ? ? ? 0

2? Kyle? ? ? ? ?0? ? ? ? ? ? 0? ? ?0? ? ? 1? ? ? ? ? 1

反對回復 2023-09-19

炎炎設計

TA貢獻1808條經驗獲得超4個贊

你可以試試這個：

n = df['Name']

df = df['Hobbies'].apply(lambda x: pd.Series([1] * len(x), index=x)).fillna(0, downcast='infer')

df.insert(0, 'Name', n)

print(df)

輸出：

Name Watch_NBA Play_PS4 Play_hockey Read Sleep

0 Paul 1 1 0 0 0

1 Jeff 0 1 1 1 0

2 Kyle 1 0 0 0 1

反對回復 2023-09-19

4 回答
0 關注
160 瀏覽

關注

添加回答

舉報

0/150

提交

取消

亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

熱搜

最近搜索清空

我需要將列列表中的每個元素轉換為 python pandas 中的新列

我需要將列列表中的每個元素轉換為 python pandas 中的新列

4 回答

添加回答