Question

我尝试从DataFrame的列子集中检索字符串，将字符串连接成一个字符串，然后将它们放入列表中，

# row_subset is a sub-DataFrame of some DataFrame
sub_columns = ['A', 'B', 'C']
string_list = [""] * row_subset.shape[0]
for x in range(0, row_subset.shape[0]):
    for y in range(0, len(sub_columns)):
        string_list[x] += str(row_subset[sub_columns[y]].iloc[x])

所以结果就像，

['row 0 string concatenation','row 1 concatenation','row 2 concatenation','row3 concatenation']

我想知道最好的方法是什么，更有效率？

Answer 1

我认为您需要先按[]按子集选择列，然后sum，或者如果需要使用分隔符join：

df = pd.DataFrame({'A':list('abcdef'),
                   'B':list('qwerty'),
                   'C':list('fertuj'),
                   'D':[1,3,5,7,1,0],
                   'E':[5,3,6,9,2,4],
                   'F':list('aaabbb')})

print (df)
   A  B  C  D  E  F
0  a  q  f  1  5  a
1  b  w  e  3  3  a
2  c  e  r  5  6  a
3  d  r  t  7  9  b
4  e  t  u  1  2  b
5  f  y  j  0  4  b

sub_columns = ['A', 'B', 'C']
print (df[sub_columns].sum(axis=1).tolist())
['aqf', 'bwe', 'cer', 'drt', 'etu', 'fyj']

print (df[sub_columns].apply(' '.join, axis=1).tolist())
['a q f', 'b w e', 'c e r', 'd r t', 'e t u', 'f y j']

非常相似的numpy解决方案：

print (df[sub_columns].values.sum(axis=1).tolist())
['aqf', 'bwe', 'cer', 'drt', 'etu', 'fyj']

pandas dataframe从列的子集连接字符串并将它们放入列表中

1 个答案: