Question

我具有以下格式的数据框：

id, ref
101, [{'id': '74947', 'type': {'id': '104', 'name': 'Sales', 'inward': 'Sales', 'outward': 'PO'}, 'inwardIssue': {'id': '76560', 'key': 'Prod-A'}}]
102, [{'id': '74948', 'type': {'id': '105', 'name': 'Return', 'inward': 'Return Order', 'outward': 'PO'}, 'inwardIssue': {'id': '76560', 'key': 'Prod-C'}}, 
      {'id': '750001', 'type': {'id': '342', 'name': 'Sales', 'inward': 'Sales', 'outward': 'PO'}, 'inwardIssue': {'id': '76560', 'key': 'Prod-X'}}]
103, [{'id': '74949', 'type': {'id': '106', 'name': 'Sales', 'inward': 'Return Order', 'outward': 'PO'}, 'inwardIssue': {'id': '76560', 'key': 'Prod-B'}},
104, [{'id': '67543', 'type': {'id': '106', 'name': 'Other', 'inward': 'Return Order', 'outward': 'PO'}, 'inwardIssue': {'id': '76560', 'key': 'Prod-BA'}}]

我正在尝试提取具有name = Sales的行并返回以下输出：

101, Prod-A
102, Prod-X
103, Prod-B

如果键值对首次出现，我就可以提取所需的数据，但如果不是id = 102的情况下不是第一次，我就不能提取

df['names'] = df['ref'].str[0].str.get('type').str.get('name')
df['value'] = df['ref'].str[0].str.get('inwardIssue').str.get('key')
df['output'] = np.where(df['names'] == 'Sales', df['value'], 0)

目前，我只能获取id = 101, 103的值

Answer 1

让我们做explode

s=pd.DataFrame(df.ref.explode().tolist())
s=s.loc[s.type.str.get('name').eq('Sales'),'inwardIssue'].str.get('key')
dfs=df.join(s,how='right')
    id                                                ref inwardIssue
0  101  [{'id': '74947', 'type': {'id': '104', 'name':...      Prod-A
2  103  [{'id': '74949', 'type': {'id': '106', 'name':...      Prod-X
3  104  [{'id': '67543', 'type': {'id': '106', 'name':...      Prod-B

Answer 2

如果您已经具有该格式的数据框，则可以将其转换为json格式，然后使用pd.json_normalize将原始df转换为一个平面数据框，然后在该平面数据框上切片/填充。 >

df1 = pd.json_normalize(df.to_dict(orient='records'), 'ref')

此平面数据帧df1

的输出

Out[83]:
       id type.id type.name   type.inward type.outward inwardIssue.id  \
0   74947     104     Sales         Sales           PO          76560
1   74948     105    Return  Return Order           PO          76560
2  750001     342     Sales         Sales           PO          76560
3   74949     106     Sales  Return Order           PO          76560
4   67543     106     Other  Return Order           PO          76560

  inwardIssue.key
0          Prod-A
1          Prod-C
2          Prod-X
3          Prod-B
4         Prod-BA

最后，在df1上切片

df_final = df1.loc[df1['type.name'].eq('Sales'), ['type.id', 'inwardIssue.key']]

Out[88]:
  type.id inwardIssue.key
0     104          Prod-A
2     342          Prod-X
3     106          Prod-B

Pandas-根据某些键值（不按顺序）从Dataframe中提取值

2 个答案: