python - 如果数据框中的列为空，如何创建条件子句？

Question

我有一个看起来像这样的df：

fname                   lname
joe                      smith
john smith      
[e-mail]            
jacky /jax               jack
[e-mail]                  non
john (jack)              smith
Bob J. Smith

我想创建一个逻辑，说如果lname是空的，如果有两个OR三个字符串fname分开第二个字符串OR第三个字符串并将其推入lname列。如果电子邮件地址保持fname原样，并且fname列中有斜杠或括号，并且保持原样没有值lname。

新的df：

fname                   lname
    joe                      smith
    john                     smith
    [e-mail]            
    jacky /jax               jack
    [e-mail]                  non
    john (jack)              smith
    Bob J.                   smith

到目前为止分隔两个字符串的代码：

df[['lname']] = df['name'].loc[df['fname'].str.split().str.len() == 2].str.split(expand=True)

标签： pythonpython-3.xpandas

Answer 1

使用以下示例数据框：

df = pd.DataFrame({'fname': ['joe', 'john smith', '[e-mail]', 'jacky /jax', '[e-mail]', 'john (jack)', 'Bob J. Smith'],
     'lname': ['smith', '', '', 'jack', 'non', 'smith', '']})

您可以使用np.where()：

conditions = (df['lname']=='') & (df['fname'].str.split().str.len()>1)

df['lname'] = np.where(conditions, df['fname'].str.split().str[-1].str.lower(), df['lname'])

产量：

           fname  lname
0            joe  smith
1     john smith  smith
2  [e-mail]       
3     jacky /jax   jack
4        [e-mail]    non
5    john (jack)  smith
6   Bob J. Smith  smith

要从填充fname了其列的行的列中删除最后一个字符串：lname

df['fname'] = np.where(conditions, df['fname'].str.split().str[:-1].str.join(' '), df['fname'])

产量：

           fname  lname
0            joe  smith
1           john  smith
2  [e-mail]       
3     jacky /jax   jack
4        [e-mail]    non
5    john (jack)  smith
6         Bob J.  smith