How can i merge the dataframe which type is array?

1

for example.

   x      y
1 [1,2] [3,4]
2.[5,6] [7,8]

and… I want to get like this:

      z
1 [1,2,3,4]
2 [5,6,7,8]

Is there have some idea resolve this problem?
thank you. and sorry my poor english… T^T

2

Good morning,
You could use

torch.cat((x,y),1)

as in

import torch
import numpy as np
a = [[1,2],[5,6]]
b = [[3,4],[7,8]]
x = torch.from_numpy(np.array(a))
y = torch.from_numpy(np.array(b))
u = torch.cat((x,y),1)
print(u)

1 Like
3

Oh… your answer is also correct. but If the csv file contain “list” type data

I should read the file like below:

df = pd.read_csv(“train_after.csv”, converters={“data”: literal_eval, “label”: literal_eval})

Thank you:)