stackoverflow python笔记1，按条件合并dataframe中字符

Question:

I have the following dataframe

first_char second_char type

1          a           b  1/1
2          a           b  0/1
3          a           b  0/1
4          c           d  1/1
5          c           d  0/1
6          c           d  0/0

I would like to combine these columns into one as such:

1       bb
2       ab
3       ab
4       dd
5       cd
6       cc

Answer 1

ind = df['type'].str.split('/', expand=True).astype(int).to_numpy()
arr2 = df[['first_char','second_char']].to_numpy()    

df['new'] = arr2[np.arange(ind.shape[0])[:,None], ind].sum(1)
print (df)
  first_char second_char type new
1          a           b  1/1  bb
2          a           b  0/1  ab
3          a           b  0/1  ab
4          c           d  1/1  dd
5          c           d  0/1  cd
6          c           d  0/0  cc

Answer 2

s = df["type"].str.split("/", expand=True).astype(int)

df["new"] = np.take_along_axis(df[["first_char","second_char"]].to_numpy(),s.to_numpy(), axis=1).sum(1)

print (df)

  first_char second_char type new
1          a           b  1/1  bb
2          a           b  0/1  ab
3          a           b  0/1  ab
4          c           d  1/1  dd
5          c           d  0/1  cd
6          c           d  0/0  cc

Answer 3

df.apply(lambda x: ''.join([(x['first_char']+x['second_char'])[int(number)] for number in x['type'].split('/')]),axis=1)

1    bb
2    ab
3    ab
4    dd
5    cd
6    cc