无法根据 pandas 数据帧中的条件为值添加前缀

Question

首先，我是Python的新手，所以我不是专家...

这是我的问题。我有这个数据帧：

   CODE_IRIS  PDL_RESIDENTIEL  PDL_TOTAL  CONSO_RESIDENTIEL  CONSO_TOTALE
0  10040101               500        510              11264         26677
1  10040102               806        809              16234         17318
2  10040201               921        925              14451         17065
3  10040202               937        943              13036         19516
4  10049999                94         94               1287          1287

问题是，CODE_IRIS是一个object，应该有 9 个字符，如下所示：

       CODE_IRIS  PDL_RESIDENTIEL  PDL_TOTAL  CONSO_RESIDENTIEL  CONSO_TOTALE
17861  766810113              588        593               9344         14743

因此，当 CODE_IRIS 中的字符长度低于 9 时，我需要传递前缀0，就像我在 Excel 上使用公式 =IF(LEN([@[Code IRIS]]) < 9; 0&[@[Code IRIS]]; [@[Code IRIS]])

所做的那样。

现在，当我尝试找到只有 8 个字符elec.loc[elec['CODE_IRIS'].str.len() < 9]的值时，我得到的结果是：

Out[393]: 
Empty DataFrame
Columns: [CODE_IRIS, PDL_RESIDENTIEL, PDL_TOTAL, CONSO_RESIDENTIEL, CONSO_TOTALE]
Index: []

然后当我尝试查看每个值与 elec['CODE_IRIS'].str.len() 的长度时，我得到的结果是：

Out[396]: 
0       NaN
1       NaN
...
Name: CODE_IRIS, Length: 23905, dtype: float64

虽然列CODE_IRIS绝对是一个object，正如你在这里看到的：

elec.info()
<class 'pandas.core.frame.DataFrame'>
Int64Index: 23905 entries, 0 to 23904
Data columns (total 5 columns):
CODE_IRIS            23905 non-null object
PDL_RESIDENTIEL      23905 non-null int64
PDL_TOTAL            23905 non-null int64
CONSO_RESIDENTIEL    23905 non-null int64
CONSO_TOTALE         23905 non-null int64
dtypes: int64(4), object(1)
memory usage: 1.1+ MB

我不明白。有人可以解释我出了什么问题吗？

(我希望我已经使自己尽可能容易理解(。谢谢！

Answer 1

您可以在每个CODE_IRIS上使用zfill：

df['CODE_IRIS'] = df['CODE_IRIS'].map(lambda x: str(x).zfill(9))