Maison >développement back-end >Tutoriel Python >Comment extraire des éléments d'un tableau en fonction des indices d'un autre tableau à l'aide de l'indexation de tableaux entiers ?

Comment extraire des éléments d'un tableau en fonction des indices d'un autre tableau à l'aide de l'indexation de tableaux entiers ?

Linda Hamilton
Linda Hamiltonoriginal
2024-11-09 02:55:02905parcourir

How to Extract Elements from an Array Based on Indices in Another Array Using Integer Array Indexing?

Utilisation de l'indexation de tableaux entiers pour extraire des éléments basés sur des indices de tableaux secondaires

Dans le scénario donné, l'objectif est de récupérer des éléments spécifiques d'un tableau A en utilisant des indices spécifiés dans un deuxième tableau B. Au lieu de s'appuyer sur np.take ou np.choose, une approche plus simple consiste à utiliser l'indexation de tableau d'entiers de NumPy :

A[np.arange(A.shape[0]),B.ravel()]

Voici comment ce code atteint l'objectif résultat souhaité :

  • np.arange(A.shape[0]) crée un tableau d'indices de 0 à (A.shape[0] - 1), représentant les lignes de A.
  • B.ravel() aplatit B en un tableau unidimensionnel, garantissant qu'il contient le même nombre d'éléments que les lignes de A.
  • L'opération d'indexation A[...] récupère les éléments de A utilisant les indices de ligne et les indices de colonne de B.

Cette approche est particulièrement utile lorsque B est un tableau 1D ou une liste d'indices de colonne. En sautant l'opération d'aplatissement, le code devient encore plus simple :

A[np.arange(A.shape[0]),B]

Exemple :

A = np.array([[0, 1], [2, 3], [4, 5]])
B = np.array([1, 0, 1])

result = A[np.arange(A.shape[0]), B]

print(result)
# Output: [1, 2, 5]

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!

Déclaration:
Le contenu de cet article est volontairement contribué par les internautes et les droits d'auteur appartiennent à l'auteur original. Ce site n'assume aucune responsabilité légale correspondante. Si vous trouvez un contenu suspecté de plagiat ou de contrefaçon, veuillez contacter admin@php.cn