python中与熊猫的父子关系

Question

我有一个超过 10000 行的 csv 文件，现在我想创建一个新列来显示父子之间的依赖关系。基于上述规则和政策，存在如下：

1. 确定并显示数据与哪个家庭相关的唯一代码团队。
2. 如果 Position 是 C ，则 Parent 字段必须满足其相对位置。李>
3. 如果 Position 是 PF，则 Child 字段必须用它的位置来满足，但如果有其他位置存在（C 和 PF 除外）它必须满足，因为它也是相对位置。
4. 存在任何其他职位，孩子必须履行。

CSV 中的现有数据：

Team        Position
Atlanta Hawks   C
Atlanta Hawks   PF
Atlanta Hawks   PG
Atlanta Hawks   SF
Atlanta Hawks   SG

结果：

Parent Child
C         PF
PF        PG
PF        SF
PF        SG

我尝试在这方面与 Panda 合作。下面的代码在第一个条件下运行良好。如果有人帮助我修改此代码，我将不胜感激？

  import pandas as pd
    
    df = pd.read_csv("C:\\Users\\Desktop\\nba.csv")
    
    gdf = df['Team']
    gdf.to_csv('C:\\Users\\Desktop\\nba-dependency.csv')
    for g in gdf:
        df.loc[df['Position']=='C','Parent']=df['Position']
        df.loc[df['Position']=='PF','Parent']=df['Position']
    
    df.to_csv('C:\\Users\\Desktop\\result.csv')

Answer 1

如果我理解正确的话，这应该是正确的做法：

import pandas as pd
import numpy as np

data = {"team":["ah","ah","ah","ah","ah"],"position":["C","PF","PG","SF","SG"]}

df = pd.DataFrame(data)

df["Parent"] = ["C" if x=="C" or x=="PF" else "PF" for x in df["position"]]
df["Child"] = df["position"]
print(df)

这里的输出将是

  team position Parent Child
0   ah        C      C     C
1   ah       PF      C    PF
2   ah       PG     PF    PG
3   ah       SF     PF    SF
4   ah       SG     PF    SG

我不完全确定第一行是否如您所愿，您可能想删除该行（您没有在结果中显示它，如果是，则将其删除），其余的看起来像您定义的那样。