如何将array（[array（[…]，type = float32）]，[array（[…]，type = float32）]）转换为array（[…]，[…]）？

Question

Here is my code:

def extract_doc2vec(doc, w2v):
vecs = [w2v[word] for word in doc if word in w2v]
if len(vecs) > 0:
    vecs = np.asarray(vecs).sum(axis=0)
    return vecs
y = newsgroups_all['target']

x= []

for doc in newsgroups_all['data']:
    edv = extract_doc2vec(list(gensim.utils.tokenize(doc)),model.wv)
    x.append(edv)

I have to convert retrain_doc which can be fit into

for train_index, test_index in sssd.split(x, y):
   X_train, X_test = x[train_index], x[test_index]
   y_train, y_test = y[train_index], y[test_index]

Answer 1

将列表x转换为数据帧，然后将其转换为numpy数组并应用shuffle拆分

data = pd.DataFrame(x).to_numpy()