如何在 Pandas 中基于依赖值更新 DataFrame？

Question

我需要根据依赖值更新数据框，应该如何操作？

比如，有一个输入数据框df：

id      dependency
10
20       30
30       40
40
50       10
60       20     

这里有：

20 -> 30和30 -> 40。因此，最终结果将是20 -> 40和30 -> 40。

同样地，60 -> 20 -> 30 -> 40，因此最终结果将是60 -> 40。

最终结果：

id      dependency   final_dependency
10
20       30            40
30       40            40
40
50       10            10
60       20            40

- bhavesh

---

非常糟糕的问题陈述。我们有...将是，我们有...将是。 - Mykola Zotko

Answer 1

你可以使用networkx来做到这一点。首先，创建一个具有依赖关系的节点图：

df_edges = df.dropna(subset=['dependency'])
G = nx.from_pandas_edgelist(df_edges, create_using=nx.DiGraph, source='dependency', target='id')

现在，我们可以找到每个节点的根祖先并将其作为新列添加：

def find_root(G, node):
    ancestors = list(nx.ancestors(G, node))
    if len(ancestors) > 0:
        root = find_root(G, ancestors[0])
    else:
        root = node
    return root

df['final_dependency'] = df['id'].apply(lambda x: find_root(G, x))
df['final_dependency'] = np.where(df['final_dependency'] == df['id'], np.nan, df['final_dependency'])

最终数据框：

   id  dependency  final_dependency
0  10         NaN               NaN
1  20        30.0              40.0
2  30        40.0              40.0
3  40         NaN               NaN
4  50        10.0              10.0
5  60        20.0              40.0

- Shaido

Answer 2

一种方法是创建自定义函数：

s = df[df["dependency"].notnull()].set_index("id")["dependency"].to_dict()

def func(val):
    if not s.get(val):
        return None
    while s.get(val):
        val = s.get(val)
    return val

df["final"] = df["id"].apply(func)

print (df)

   id  dependency  final
0  10         NaN    NaN
1  20        30.0   40.0
2  30        40.0   40.0
3  40         NaN    NaN
4  50        10.0   10.0
5  60        20.0   40.0

- Henry Yik

Answer 3

你已经有了几个答案。iterrows()是一个比较昂贵的解决方案，但我也想让你知道这个方法。

import pandas as pd

raw_data = {'id': [i for i in range (10,61,10)],
            'dep':[None,30,40,None,10,20]}
df = pd.DataFrame(raw_data)

df['final_dep'] = df.dep

for i,r in df.iterrows():

    if pd.notnull(r.dep):
        x = df.loc[df['id'] == r.dep, 'dep'].values[0]
        if pd.notnull(x):
            df.iloc[i,df.columns.get_loc('final_dep')] = x
        else:
            df.iloc[i,df.columns.get_loc('final_dep')] = r.dep

print (df)

这个的输出将是：

   id   dep final_dep
0  10   NaN       NaN
1  20  30.0        40
2  30  40.0        40
3  40   NaN       NaN
4  50  10.0        10
5  60  20.0        30

- Joe Ferndz