Dictionnaires Python - ajout de tableaux à un dictionnaire pour une clé spécifique.

Ceci est une extension du premier répondeur et mon commentaire au premier répondeur.

Si j'ai les tableaux suivants:

array1 = [1,2,3,4]
array2 = [2,6,27,1]
array3 = [6,1,2,3]
array4 = [7,1,2,6]
array5 = [8,2,1,6]
array6 = [3,4,1,1]

Mon dictionnaire est le suivant:

from collections import defaultdict
my_dict = defaultdict(list)

Maintenant, je vais ajouter les tableaux suivants pour différentes clés:

my_dict[1].append(array1)
my_dict[1].append(array2)
my_dict[1].append(array3)
my_dict[2].append(array4)
my_dict[3].append(array5)
my_dict[3].append(array6)

Contenu de my_dict

{1:[[1,2,3,4],[2,6,27,1],[6,1,2,3]], 2:[[7,1,2,6]], 3:[[8,2,1,6],[3,4,1,1]] }

Pour obtenir le tableau associé à key = 1

foo = my_dict[1]
foo
[[1,2,3,4],[2,6,27,1],[6,1,2,3]]

Maintenant, pour obtenir la 3e colonne de données:

import numpy as np #Need to use it to convert to array and float later on
foo = np.array(foo);# Have to convert it to an array
doo = foo[:,2]
doo
[3,27,2]

Maintenant, il peut y avoir des situations où votre tableau peut être un mélange de chaîne et de flottant ou entier tel que tous les éléments du tableau sont des chaînes.

foo 
[['1','2','3','4'],['2','6','27','1'],['6','1','2','3']]
foo = np.float32(foo)
foo
[[1.,2.,3.,4.],[2.,6.,27.,1.],[6.,1.,2.,3.]]

Conclusion: nous pouvons ajouter des tableaux à un dictionnaire pour une clé spécifique en utilisant le defaultdict. Des sous-ensembles de la matrice sélectionnée peuvent également être extraits via la méthode de sélection de colonne normale. Si le tableau associé à la clé contient des éléments de chaîne, nous pouvons convertir les données en numpy.float32, disons ou partout où nous devons faire un calcul avec des nombres.

Merci, Anthony de Sydney