Python如何获取2维列表中的每个第一个元素

Question

我有一个这样的列表:

a = ((4.0, 4, 4.0), (3.0, 3, 3.6), (3.5, 6, 4.8))

我想要一个像这样的结果(列表中的每个第一个元素):

4.0, 3.0, 3.5

我尝试了[:: 1] [0],但它不起作用

我刚刚开始学习Python几周前.Python版本= 2.7.9

Answer 1

您可以[0]从列表推导中的每个元素获取索引

>>> [i[0] for i in a]
[4.0, 3.0, 3.5]

也正好是迂腐,你没有list的list,你有一个tuple的tuple.

Answer 2

使用拉链

columns = zip(*rows) #transpose rows to columns
print columns[0] #print the first column
#you can also do more with the columns
print columns[1] # or print the second column
columns.append([7,7,7]) #add a new column to the end
backToRows = zip(*columns) # now we are back to rows with a new column
print backToRows

你也可以使用numpy

a = numpy.array(a)
print a[:,0]

Answer 3

你可以用这个：

a = ((4.0, 4, 4.0), (3.0, 3, 3.6), (3.5, 6, 4.8))
a = np.array(a)
a[:,0]
returns >>> array([4. , 3. , 3.5])

Answer 4

你可以得到它

[ x[0] for x in a]

这将返回每个列表的第一个元素的列表 a

Answer 5

3种方法比较

1. 2D 列表：5.323603868484497 秒
2. Numpy 库：0.3201274871826172 秒
3. Zip（感谢 Joran Beasley）：0.12395167350769043 秒

D2_list=[list(range(100))]*100
t1=time.time()
for i in range(10**5):
    for j in range(10):
        b=[k[j] for k in D2_list]
D2_list_time=time.time()-t1

array=np.array(D2_list)
t1=time.time()        
for i in range(10**5):
    for j in range(10):
        b=array[:,j]        
Numpy_time=time.time()-t1

D2_trans = list(zip(*D2_list)) 
t1=time.time()        
for i in range(10**5):
    for j in range(10):
        b=D2_trans[j]
Zip_time=time.time()-t1

print ('2D List:',D2_list_time)
print ('Numpy:',Numpy_time)
print ('Zip:',Zip_time)

Zip 方法效果最好。当我必须为未安装 numpy 的集群服务器中的 mapreduce 作业执行一些列明智的过程时，它非常有用。