Более вещий способ индексации со списком координат

Question

Поэтому я разрабатывающей программу, которую нужно индексировать вложенный список, используя сохраненный список координат:

например

coordinate = [2,1] 

Для функции, возвращает элемент в вложенного списка, я использую

return room[coordinate[0]][coordinate[1]] 

Мои инстинкты программирования говорят мне, что это, кажется, слишком долго, и там должен быть короче способ сделать это, особенно в Python, бушель Я не могу найти ничего подобного. Кто-нибудь знает, есть ли такой метод?

Answer 1

Вы можете распаковать координаты в более чем одной переменной.

i, j = [2, 1] 
return room[i][j] 

или

coordinates = [2, 1] 
### do stuff with coordinates 
i, j = coordinates 
return room[i][j] 

Answer 2

coordinates[2][1] = "blah" 

, как вы правильно индекс в вложенный список

с использованием кортежа, вероятно, хороший способ для хранения статического, неизменных координаты

myCoord = (2,1) 

Вашего пути индексации в ваш вложенный room массив выглядит правильно и является удобочитаемым. Мне нужно будет узнать больше о том, как вы его используете, чтобы рекомендовать, какие типы данных использовать.

редактировать
В ответ на ваш комментарий, я бы сказал, что если бы это была функция, принимать x и y как входы или, если это невозможно сделать x,y = myTuple
так, что вы могли бы сделать это :

room[x][y] 

вместо

room[coords[0]][coords[1]] 

это сделает его бит более читабельный, так как это, кажется, ваша забота. Существует не способ изначально индекса вложенного списка, используя кортеж

Answer 3

Вы можете определить свою собственную функцию рекурсивного индексирования:

def rec(x, i): 
    if i: return rec(x[i[0]], i[1:]) 
    else: return x 

Что дает:

>>> room = [[[1,2,3],[4,5,6]],[[7,8,9],[10,11,12]],[[13,14,15],[16,17,18]]] 
>>> rec(room, (2,1)) 
[16, 17, 18] 
>>> rec(room, [2,1,1]) 
17 

Answer 4

Модуль numpy имеет удобное индексирование , Это будет хорошо работать, если ваш room очень большой.

>>> import numpy as np 
>>> room = np.arange(12).reshape(3,4) 
>>> room 
array([[ 0, 1, 2, 3], 
     [ 4, 5, 6, 7], 
     [ 8, 9, 10, 11]]) 
>>> coords = (2, 1) # it's important this is a tuple 
>>> room[coords] 
9 

Чтобы преобразовать room переменную в numpy массив, предполагая, что это 2 мерных вложенный список, просто сделать

>>> room = [[0, 1, 2, 3, 4], 
      [0, 1, 2, 3, 4], 
      [0, 1, 2, 3, 4], 
      [0, 1, 2, 3, 4], 
      [0, 1, 2, 3, 4]] 
>>> room = np.array(room) 
array([[0, 1, 2, 3, 4], 
     [0, 1, 2, 3, 4], 
     [0, 1, 2, 3, 4], 
     [0, 1, 2, 3, 4], 
     [0, 1, 2, 3, 4]]) 

Answer 5

Простое решение было бы использовать NumPy:

In [1]: import numpy 

In [2]: a = numpy.arange(15).reshape(3, 5) 

In [3]: a 
Out[3]: 
array([[ 0, 1, 2, 3, 4], 
     [ 5, 6, 7, 8, 9], 
     [10, 11, 12, 13, 14]]) 

In [4]: coords = (2, 1) 

In [5]: a[coords] 
Out[5]: 11 

Если это не вариант, вы можете создать подкласс список для реализации такого рода индексации:

class MyList(list): 
    def __getitem__(self, index): 
     if isinstance(index, collections.Iterable): 
      return reduce(operator.getitem, index, self) 
     return list.__getitem__(self, index) 

Пример использования:

>>> a = MyList([[ 0, 1, 2, 3, 4], 
       [ 5, 6, 7, 8, 9], 
       [10, 11, 12, 13, 14]]) 
>>> coords = (2, 1) 
>>> a[coords] 
11 

Answer 6

Основная проблема заключается в том, что вы используете список в качестве структуры данных для информации, к которой вы хотите получить доступ нестандартным способом.

Сам список прекрасно, вы, возможно, в конечном итоге захотите сделать его объектом, имеющим список как хранилище, но представляющим собой более ориентированный на задачи интерфейс пользователя.

Если вы не готовы к решению класса, это становится вам ближе к Не повторяйте себе:

def room_at_coordinate(rooms, coordinate) 
    return rooms[coordinate[0]][coordinate[1]] 

>>> room_at_coordinate(rooms, coordinate) 
'bathroom' 

Эта функция будет скользить естественно в объект, если вы решили идти по этому пути.