Python学习笔记：分割取元素

类似于另外一篇文所讲，仅记录一下代码备用。

Python学习笔记：给定邮编，判断区域

1.需求

提取第1个元素。

2.实操

import pandas as pd
df = pd.DataFrame({
    'name':['[abc,123,456]','[abcd,123,456]','[1bcasdf,123,456]','[ppp]']
    })
df
'''
                name
0      [abc,123,456]
1     [abcd,123,456]
2  [1bcasdf,123,456]
3              [ppp]
'''

• 方法一

result = []
for i in df['name']:
    if len(i.split(',')) > 1:
        # print(i)
        result.append(i.split(',')[0][1:]) # 多个元素
    else:
        result.append(i.split(',')[0][1:-1]) # 只有1个元素
result

• 方法二

df['name_new'] = df['name'].apply(lambda x: x.replace('[','').replace(']',''))
df['name_new2'] = df['name_new'].str.split(',', expand=True)[0]

利用 expand 参数对分割后的数据进行展开。