我有一个包含两列def find_ages(*args):
if args[0] == 'Anna':
# Return Anna's data
return 18
elif args[0] == ' Ben':
# Return Ben's data
return 45
@patch.object(get_ages, 'get_age', side_effect=find_ages)
def test_get_ages(self, get_age):
和col1
的df。我正在尝试拆分是否有逗号分隔的值。
例如:从col2
到1,2 - mason
和1 - mason
df
2 - mason
我的例外输出:
df= pd.DataFrame(data = {'col1' : ['1,2', '2', '3', '4', '5,6,7,8'], 'col2' : ['mason', 'marcus', 'anthony', 'paul', 'bruno']})
col1 col2
0 1,2 mason
1 2 marcus
2 3 anthony
3 4 paul
4 5,6,7,8 bruno