根据列表中的元素在 dataframe 中添加行

Question

我有以下列表 & dataframe：

lis = ["Color", "Material", "Handle"]

Code for dataframe:

data = [[1, 'Color', 'Yellow', 'SourceA'],
        [1, 'Material', 'Plastic', 'SourceA'],
       [1, 'Handle', 'Y', 'SourceB'],
       [2, 'Color', 'Blue', 'SourceB'],
       [2, 'Handle', 'N', 'SourceA'],
       [3, 'Color', 'Black', 'SourceA'],
       [3, 'Color', 'Black', 'SourceB'],
       [3, 'Material', 'Steel', 'SourceA']]
  
df_one = pd.DataFrame(data, columns=['Id', 'feature', 'feature_value', 'Source'])

df_one = 

| ID | feature  | feature_value | Source  |
| 1  | Color    | Yellow        | SourceA |
| 1  | Material | Plastic       | SourceA |
| 1  | Handle   | Y             | SourceB |
| 2  | Color    | Blue          | SourceB |
| 2  | Handle   | N             | SourceA |
| 3  | Color    | Black         | SourceA |
| 3  | Color    | Black         | SourceB |
| 3  | Material | Steel         | SourceA |

我需要每个 ID 都具有“lis”中列出的所有功能。 ID1 有“颜色”、“材质”、“手柄”，但 ID2 没有“材质”，ID3 没有“手柄”。 我需要我的 output 如下所示：

| ID | feature  | feature_value | Source  |
| 1  | Color    | Yellow        | SourceA |
| 1  | Material | Plastic       | SourceA |
| 1  | Handle   | Y             | SourceB |
| 2  | Color    | Blue          | SourceB |
| 2  | Handle   | N             | SourceA |
| 2  | Material | null          | UNK     |
| 3  | Color    | Black         | SourceA |
| 3  | Color    | Black         | SourceB |
| 3  | Material | Steel         | SourceA |
| 3  | Handle   | null          | UNK     | 

我尝试遍历 dataframe 的行并为每一行创建一个字典，但是因为 ID 列不是唯一的，所以我没有键和值的唯一键。

任何帮助，将不胜感激！

Answer 1

如果您确定至少有一个ID具有lis的三个值， DataFrame.complete适合您。

#pip install pyjanitor
import janitor
​
out = df_one.complete("Id", "feature", fill_value={"Source": "UNK"})
​

Output：

df_one.groupby("Id")["feature"].agg(set).eq((lis)).any()
#True

print(out)

   Id   feature feature_value   Source
0   1     Color        Yellow  SourceA
1   1  Material       Plastic  SourceA
2   1    Handle             Y  SourceB
3   2     Color          Blue  SourceB
4   2  Material           NaN      UNK
5   2    Handle             N  SourceA
6   3     Color         Black  SourceA
7   3     Color         Black  SourceB
8   3  Material         Steel  SourceA
9   3    Handle           NaN      UNK