MATLAB ind2sub équivalent en Python

Une recherche sur Google me conduire à ce lien: https://github.com/jjakeman/pyheat/blob/master/utilities/math_utils.py

De ce que je peux dire, il n'y a pas de diriger la mise en œuvre de ces fonctions de MATLAB.

---

S'avère juste, je ne peux pas lire la documentation correctement. Si vous souhaitez que la fonctionnalité de sub2ind, vous aurez besoin de la ravel_multi_index fonction. La déclaration de la fonction indique que vous avez besoin de deux entrées. La première entrée est un 2D numpy tableau où chaque ligne sont les endroits pour une dimension particulière. Par exemple, si vous souhaitez appliquer ind2sub sur une matrice 2D, vous pouvez spécifier un 2D numpy tableau dans lequel la première ligne se compose de tous les emplacements de lignes que vous voulez, et la deuxième ligne se compose de tous les emplacements de colonne que vous voulez. La deuxième entrée est tuple qui détermine la taille de chaque dimension, donc pour un tableau 2D, ce serait le nombre de lignes et de colonnes.

Faire ind2sub, vous voulez le unravel_index fonction. La première entrée est un tableau linéaire des indices que vous voulez convertir dans des endroits de chaque dimension dans votre tableau. La deuxième entrée est un n-uplet de dimensions, comme précédemment.

Je vais laisser le post en bas pour la postérité, dans le cas où vous voulez essayer et de mettre en œuvre vous-même.

---

Cependant, vous pouvez certainement mettre en œuvre ces vous-même. Je suppose que parce que vous avez attaché votre post avec numpy que vous aurez envie d'une numpy-esque solution. Rappelez-vous, dans numpy vous accéder à des éléments de ligne majeur, pas colonne majeur, et donc, étant donné deux ensembles tels que l'un est pour la ligne et l'autre pour la colonne des indices (0-indexé), sub2ind pour les matrices 2D est très simple:

def sub2ind(array_shape, rows, cols):
    return rows*array_shape[1] + cols

array_shape est un tableau de deux éléments dont le premier élément est le nombre de lignes dans la matrice et le second élément est le nombre de colonnes. Si vous vous souvenez, vous pouvez accéder à un élément dans une rangée-major de la matrice par:

ind = r*cols + c

(r,c) sont la ligne et la colonne de l'index que vous voulez, à condition de la 0-indexé. De faire le chemin inverse, vous pouvez utiliser division entière et le module:

def ind2sub(array_shape, ind):
    rows = (ind.astype('int') / array_shape[1])
    cols = (ind.astype('int') % array_shape[1]) # or numpy.mod(ind.astype('int'), array_shape[1])
    return (rows, cols)

Ici, la sortie est à deux éléments d'un tuple dont le premier élément est la rangée d'emplacements et le deuxième élément est la colonne endroits. Pour résumer ind2sub, pour accéder à la ligne que vous voulez, vous prenez le linéaire de l'indice et de faire une division entière avec les colonnes. Pour obtenir la colonne que vous souhaitez, vous trouverez le module ou le reste. Aller à 3 dimensions et au-delà est un peu plus compliqué. Je vais vous laisser jeter un oeil au lien que j'ai mentionné ci-dessus pour plus de détails.

Évidemment, je ne place pas de contrôle d'erreur dans les fonctions ci-dessus, de sorte que vous feriez bien évidemment utiliser array_shape à votre avantage dans ce cas. Une meilleure façon de faire ce que vous voulez serait quelque chose comme:

def sub2ind(array_shape, rows, cols):
    ind = rows*array_shape[1] + cols
    ind[ind < 0] = -1
    ind[ind >= array_shape[0]*array_shape[1]] = -1
    return ind

def ind2sub(array_shape, ind):
    ind[ind < 0] = -1
    ind[ind >= array_shape[0]*array_shape[1]] = -1
    rows = (ind.astype('int') / array_shape[1])
    cols = ind % array_shape[1]
    return (rows, cols)

J'ai fait quelques erreur de base de vérification pour s'assurer qu'aucune des lignes ou des colonnes pour sub2ind ou linéaire des indices pour ind2sub sont hors limites. J'ai mis ces emplacements à -1 de sorte que vous savez que vous foiré quelque part.

Bonne chance!

OriginalL'auteur rayryeng

Bâtiment sur @rayryeng et @theblackcat de réponses, vous devez également noter que vous devrez utiliser Fortran style d'indexation, et de rappeler que Python est 0 indexé alors que MATLAB est de 1 indexés.

Le Fortran style exigence trompé moi un peu.

En Python:

np.unravel_index(7, [1, 2, 3, 4], 'F')
(0, 1, 0, 1)

et dans MATLAB/Octave

[a, b, c, d] = ind2sub([1, 2, 3, 4], 8)
a =  1
b =  2
c =  1
d =  2

OriginalL'auteur Gilly