python - Pandas 中的子字符串元素

Question

对于 python DataFrame，我需要为列中的每个元素选择特定字符

df['y'] = df['x'].apply(lambda x: x[1:4])

但出现此错误

类型错误:“NoneType”对象不可订阅

Answer 1

我相信需要indexing with str它也处理 NaN 和 None:

df = pd.DataFrame({ 'x' : ['abcdef','av', 'a', None, np.nan]})
df['y'] = df['x'].str[1:4]
print (df)

        x     y
0  abcdef   bcd
1      av     v
2       a      
3    None  None
4     NaN   NaN

编辑:

我建议首先按 str[1:] 进行索引，然后按 sparator 进行拆分，然后使用 str[0] 来查找第一个拆分列表:

df = pd.DataFrame({ 'x' : ['abc|etf','av|ee', 'ayty|s', None, np.nan]})

df['y'] = df['x'].str[1:].str.split('|').str[0]
print (df)

         x     y
0  abc|etf    bc
1    av|ee     v
2   ayty|s   yty
3     None  None
4      NaN   NaN

编辑:

df = pd.DataFrame({ 'x' : ['abc|Ttf','av|Re', 'ayty|s','ayty|','ayty', None, np.nan]})

def f(x):
    #find position
    pos = x.find('|')
    #if no match return same value
    if pos == -1:
        return x
    #else procees code - split
    else:
        return x[:pos+2]

mask = df['x'].notnull()
df.loc[mask, 'y'] = df.loc[mask, 'x'].apply(f)
print (df)

         x       y
0  abc|Ttf   abc|T
1    av|Re    av|R
2   ayty|s  ayty|s
3    ayty|   ayty|
4     ayty    ayty
5     None     NaN
6      NaN     NaN