在列上迭代并用提取的字符串替换值[Pandas]

Question

我有一个数据集，看起来是这样的：

  A   B
1 aa  1234
2 ab  3456
3 bc  [1357, 2468]
4 cc  8901
...

我需要遍历B列，并将方括号（[]）中的所有值替换为这些括号中的四个左数字，因此数据集将如下所示：

  A   B
1 aa  1234
2 ab  3456
3 bc  1357
4 cc  8901
...

我有这个密码：

for item in df['B']:
if len(item) > 4:
    item_v = str(item[1:5])
    df['B'][item] = item_v 
    print(df['B'][item])

但是，如果我检查df的头，它仍然有旧的值：

   > df['B'].head()

   >  A   B
    1 aa  1234
    2 ab  3456
    3 bc  [1357, 2468]
    4 cc  8901
    ...

我做错什么了？

Answer 1

最简单和最快的方法是使用Pandasstr.get()函数并为所需的结果创建另一列。

解决方案#1 如果B中的值是整数[1234,3456,[1357, 2468],8901]，则第一个解决方案有效

df['C'] = df['B'].str.get(0).astype(float)
df.C.fillna(df['B'], inplace=True)
df['C'] = df.C.astype(int, inplace=True)

输出：

A             B     C
0  aa          1234  1234
1  ab          3456  3456
2  bc  [1357, 2468]  1357
3  cc          8901  8901

然后，如果不需要，可以删除列B。

解决方案#2 如果B中的值是字符串['1234','3456',['1357', '2468'],'8901']，则此解决方案有效

import re
df['digits'] = df['B'].apply(lambda x: re.findall('\d+', str(x)))
df['digits'] = df['digits'].str.get(0)
print(df)

输出：

   A             B    digits
0  aa          1234   1234
1  ab          3456   3456
2  bc  [1357, 2468]   1357
3  cc          8901   8901

同样，如果不需要，可以删除列B。