python - 为 Pandas 中的新列赋值

Question

您好，我有一个数据框，如下所示:

Col1
ABC
ABCD
XYZ Ju21

我想将值分配给此数据框中的新列:

如果 Col1 不包含空格，则将 Col1(行值)复制到 Col2

如果 Cole 包含空格，则提取第一个空格之前的字符串并将其分配给 Col2

可以使用正则表达式或任何 Pandas 函数来完成吗？任何帮助都感激不尽。

Answer 1

这是结合str.contains的解决方案, str.split和 lambda内np.where :

import pandas as pd
import numpy as np
df1 = pd.DataFrame({'index':[1,2,3],'Col1':['ABC','ABCD','XYZ Ju21']})

df1['Col2'] = np.where(df1['Col1'].str.contains(' '),df1['Col1'].str.split(' ').apply(lambda x: x[0]),df1['Col1'])
print(df1)

输出:

   index      Col1  Col2
0      1       ABC   ABC
1      2      ABCD  ABCD
2      3  XYZ Ju21   XYZ