Si j'ai un tableau numpy qui contient n éléments. Comment puis-je créer une liste en choisissant des plages spécifiques dans ce tableau ?
array_input = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
A = array_input[]
B = array_input[]
C = array_input[]
Je veux savoir ce que je dois taper entre parenthèses pour que la sortie soit :
A = [1, 2, 3]
B = [4, 5, 6]
C = [7, 8, 9]
Si vous avez un objet numpy.ndarray pour commencer, pourquoi ne pas simplement travailler avec cela ? Remodelez-le :
>>> arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> arr = arr.reshape(3,-1)
>>> arr
array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
>>> arr[0]
array([1, 2, 3])
Et cela conviendrait probablement à vos besoins. Si vous insistez pour créer trois listes distinctes, vous pouvez simplement faire quelque chose comme :
>>> A, B, C = arr.reshape(3, -1).tolist()
>>> A
[1, 2, 3]
>>> B
[4, 5, 6]
>>> C
[7, 8, 9]
Cet article est collecté sur Internet, veuillez indiquer la source lors de la réimpression.
En cas d'infraction, veuillez [email protected] Supprimer.
laisse moi dire quelques mots