Python：在DataFrame中，如何查找某一列中的字符串出現在另一列中的年份？

Question

我有一個數據框，想要遍歷c2列中的所有字符串，並打印該字符串及其出現在c2列中的年份，然后還打印出出現在c1列中的第一年 （如果它存在於c1中）。 然后在另一列中計算年份之間的差異。 c2中有NaN值。

df示例：

id   year     c1                c2
0    1999     luke skywalker    han solo
1    2000     leia organa       r2d2
2    2001     han solo          finn
3    2002     r2d2              NaN
4    2004     finn              c3po
5    2002     finn              NaN
6    2005     c3po              NaN

示例打印結果：

c2            year in c2   year in c1     delta
han solo      1999         2001           2
r2d2          2000         2002           2
finn          2001         2004           3
c3po          2004         2005           1

我正在將Jupyter Notebook與python和pandas一起使用。 謝謝！

Answer 1

您可以按照以下步驟進行操作：

df1 = df[df.c2.notnull()].copy()

s = df.groupby('c1')['year'].first()
df1['year in c1'] = df1.c2.map(s)

df1 = df1.rename(columns={'year':'year in c2'})

df1['delta'] = df1['year in c1'] - df1['year in c2']

print(df1[['c2','year in c2','year in c1', 'delta']])

輸出：

         c2  year in c2  year in c1  delta
0  han solo        1999        2001      2
1      r2d2        2000        2002      2
2      finn        2001        2004      3
4      c3po        2004        2005      1

Answer 2

這是一種方法。

df['year_c1'] = df['c2'].map(df.groupby('c1')['year'].agg('first'))\
                        .fillna(0).astype(int)

df = df.rename(columns={'year': 'year_c2'})
df['delta'] = df['year_c1'] - df['year_c2']

df = df.loc[df['c2'].notnull(), ['id', 'year_c2', 'year_c1', 'delta']]

#    id  year_c2  year_c1  delta
# 0   0     1999   2001.0      2
# 1   1     2000   2002.0      2
# 2   2     2001   2004.0      3
# 4   4     2004   2005.0      1

說明

• 將c1映射到year ，按“ first”進行聚合。
• 使用此映射在c2上計算year_c1 。
• 計算delta為year_c2和year_c1之間的差。
• 刪除c2和order列中具有null行。