Python - map()分割并获得第一部分

Question

我有一本csv通过大熊猫阅读的书read_csv,

data = pandas.read_csv(file)

现在一小部分是这样的,

data['interest']:

one.a
two.a
three.a
four.b

等等...

我想创建一个新的pandas系列,比方说,

data['i']它只包含第一部分data['interest'],即

one
two
three
four

现在我可以遍历整个事情split,但是我可以用地图做吗？

像 - data['i'] = map(split_and_get_first_part, data['interest'])？

Answer 1

你可以使用这个map方法:

data['interest'].map(lambda x:x.split('.', 1)[0])

import pandas as pd

data = pd.DataFrame({'interest':['one.a', 'two.a', 'three.a', 'four.b']})
data['i'] = data['interest'].map(lambda x:x.split('.', 1)[0])
print(data)

产量

  interest      i
0    one.a    one
1    two.a    two
2  three.a  three
3   four.b   four