在熊猫中按位置提取数字？

Question

我有一个df：

                  col1
0       01139290201001
1       01139290101001
2       01139290201002
3       01139290101002
4       01139290201003
5       01139290101003
6       01139290201004
7       01139310101001
8       01139290201005
9       01139290301001
            ...      
5908      01139ÅÊ21020
5909      01139ÅÊ21013
5910      01139ÅÊ11008
5911      01139ÅÊ21011
5912      01139ÅÊ03003

仅在int情况下，我需要将前7个数字提取到新列中；在包含字符的情况下，我需要将前5和8,9个数字提取到新列中。

我尝试将此代码处理到一个组合的数据框中，以尝试解决该问题的方法，并且可以正常工作，但是当我在实际数据集上尝试该代码时，它并没有按预期工作，主要原因是我的实际{{1} }具有整数，并对其进行计算。

df

为什么还会导致NaN值？

我希望它看起来像这样：

df['col2']=df[col1][0:5]+df['col1'][8]


0       0113929020100101139290201005
1       0113929010100101139290201005
2       0113929020100201139290201005
3       0113929010100201139290201005
4       0113929020100301139290201005
5                                NaN
6                                NaN
7                                NaN
8                                NaN
9                                NaN

Answer 1

使用.apply

例如：

import pandas as pd
df = pd.DataFrame({"col1": ["01139290201001", "01139290101001", "01139290201002", "01139ÅÊ21020", "01139ÅÊ21013", "01139ÅÊ11008"]})
df["col2"] = df["col1"].apply(lambda x: x[:7] if x.isdigit() else x[:5]+x[9:11] )
print(df)

输出：

             col1     col2
0  01139290201001  0113929
1  01139290101001  0113929
2  01139290201002  0113929
3    01139ÅÊ21020  0113921
4    01139ÅÊ21013  0113921
5    01139ÅÊ11008  0113911