python - 如何使用csv和pandas制作产品代码列表？

Question

我是 Pandas 新手，我有一个 csv 文件，看起来像这样，但有 1000 行:

     type       size    obs      code
0    inox       1        hard
1    inox       2        soft
2    inox       4        soft2
3    gold       4        hard
4    silver     1        hard
5    gold       2        hard

我希望代码能够识别身份不锈钢，金银，如果软，硬等，则排序不锈钢，排序金。

   inox code 01 |
   gold code 02 |
   silver code 03 |
   hard 001 |
   soft 002 |
   soft2 003 |

我在网上没有找到任何东西，甚至在 stackoverflow 上也没有找到，我需要的导出:

        type       size     obs      code
   0    inox       1        hard     01.001.01
   1    inox       2        soft     01.002.02
   2    inox       4        soft2    01.003.03
   3    gold       4        hard     02.001.01
   4    gold       1        hard     02.001.02
   5    silver     2        soft     03.002.01

我可以使用 pandas.loc 吗？

Answer 1

使用map和cumcount

  df['type'].map({'inox': '01', 'gold': '02', 'silver': '03'})  \
+ '.' \
+ df['obs'].map({'hard': '001', 'soft': '002', 'soft2': '003'}) \
+ '.0' \
+ df.groupby(['type']).cumcount().add(1).astype(str)

<小时/>

0    01.001.01
1    01.002.02
2    01.003.03
3    02.001.01
4    03.001.01
5    02.001.02
dtype: object