python - 拆分存储在数组中的字符串值

Question

import numpy as np

import pandas as pd

a=np.array(['ma_ya','dj_kh','ja_aa'])

a1=(lambda x:x[0].split('_'),a)
a1

输出：

 (<function __main__.<lambda>>, array(['ma_ya', 'dj_kh', 'ja_aa'], 
       dtype='<U5'))

我希望输出['ma','dj','ja']使用 lambda 函数。好心的帮助

标签： pythonstringnumpylambdasplit

Answer 1

a1 = list(map(lambda x: x.split('_')[0], a))