python - 将两个不同数据帧的列强制为相同的数据类型

Question

我有两个结构如下的数据框:

print(product_combos1.head(n=5))
             product_id  count  Length
0            (P06, P09)  36340       2
1  (P01, P05, P06, P09)  10085       4
2            (P01, P06)  36337       2
3            (P01, P09)  49897       2
4            (P02, P09)  11573       2

print(testing_df.head(n=5))
                     product_id  Length
transaction_id                         
001                       [P01]       1
002                  [P01, P02]       2
003             [P01, P02, P09]       3
004                  [P01, P03]       2
005             [P01, P03, P05]       3

如何强制 testing_df 的“product_id”列，使其与 product_combos1 df 中的列格式相同？ (即 - 在括号中而不是方括号中)

Answer 1

Python 元组显示在括号中。列表显示在括号中。

更改数据框

testing_df['product_id'] = testing_df['product_id'].apply(tuple)
testing_df 

                     product_id  Length
transaction_id                         
1                        (P01,)       1
2                    (P01, P02)       2
3               (P01, P02, P09)       3
4                    (P01, P03)       2
5               (P01, P03, P05)       3

制作副本

testing_df.assign(product_id=testing_df.product_id.apply(tuple))

                     product_id  Length
transaction_id                         
1                        (P01,)       1
2                    (P01, P02)       2
3               (P01, P02, P09)       3
4                    (P01, P03)       2
5               (P01, P03, P05)       3

<小时/>

当然，除非这些实际上是字符串。然后只需将方括号替换为圆括号即可。

testing_df.assign(product_id=testing_df.product_id.str.replace('\[(.*)\]', r'(\1)'))

                     product_id  Length
transaction_id                         
1                         (P01)       1
2                    (P01, P02)       2
3               (P01, P02, P09)       3
4                    (P01, P03)       2
5               (P01, P03, P05)       3