3

我必须将值列表更改为多个数组集,如下所示:

list_train_data = [u'Class 1',
 u'Class 2',
 u'Class 3',
 u'Class 4',
 u'Class 5']

我需要将此值放入一个数组中,例如:

train_set = [['Class 1'],['Class 2'],['Class 3'],['Class 4'],['Class 5']]

如果可能,不要使用 for 循环。

4

3 回答 3

7

使用列表推导:

train_set = [[x] for x in list_train_data]

演示:

>>> list_train_data = [u'Class 1', u'Class 2', u'Class 3', u'Class 4', u'Class 5']
>>> [[x] for x in list_train_data]
[[u'Class 1'], [u'Class 2'], [u'Class 3'], [u'Class 4'], [u'Class 5']]
于 2013-10-28T11:10:42.953 回答
7

既然你添加了 numpy 标签,这里是一个 numpy 解决方案:

list_train_data = [u'Class 1',
 u'Class 2',
 u'Class 3',
 u'Class 4',
 u'Class 5']

import numpy as np
np.array(list_train_data, "O")[:, None]

结果是:

array([[u'Class 1'],
       [u'Class 2'],
       [u'Class 3'],
       [u'Class 4'],
       [u'Class 5']], dtype=object)
于 2013-10-28T11:19:48.390 回答
1

当然还有另一种不使用循环的方法,那就是使用map函数:

>>> list_train_data = ['Class 1',
                   'Class 2',
                   'Class 3',
                   'Class 4',
                   'Class 5']
>>> map(lambda x: [x], list_train_data)
[['Class 1'], ['Class 2'], ['Class 3'], ['Class 4'], ['Class 5']]

没有任何循环。然而,这比 LC 慢。但仍然没有for看到循环。

于 2013-10-28T11:40:40.703 回答