Créer matrice à partir de la liste de la liste des tuples en utilisant la compréhension liste

Question

J'ai une liste de listes de tuples list1

list1 = [[('a',0.01),('b',0.23),('c',1e-7)], 
     [('a',0.91),('b',0.067),('c',0.38)]] 

et je veux créer une matrice numpy où chaque ligne serait la deuxième valeur d'un tuple list1. Ainsi, la matrice, permet de l'appeler A, aurait forme

A = [[0.01,0.23,1e-7],[0.91,0.067,0.38]] 
A.shape 
>>> (2,3) 

Jusqu'à présent, je suis parvenu à atteindre cet objectif d'une manière lente et inefficace

A = [] 
for i in range(len(list1)): 
    A.append(np.array([v for k,v in list1[i]])) 
A = np.array(A) 

Comment puis-je faire cela en utilisant la compréhension de la liste?

Answer 1

Vous avez besoin comprehensions liste imbriquée pour cela:

np.array([[tup[1] for tup in lst] for lst in list1]) 
Out: 
array([[ 1.00000000e-02, 2.30000000e-01, 1.00000000e-07], 
     [ 9.10000000e-01, 6.70000000e-02, 3.80000000e-01]]) 

Une meilleure solution serait:

np.array(list1)[:,:,1].astype('float') 
Out: 
array([[ 1.00000000e-02, 2.30000000e-01, 1.00000000e-07], 
     [ 9.10000000e-01, 6.70000000e-02, 3.80000000e-01]])