Список индексов с другим списком? - Python 2.7

Question

Я немного новичок в python (переключение с IDL), поэтому я извиняюсь, если не использую правильную терминологию. Я пробовал искать аналогичные questions, но, похоже, не понял. У меня есть два списка, и я пытаюсь создать гистограмму данных, где dat2 = 1. Я пытался делать это несколько способов, и он продолжает давать мне TypeError

import matplotlib.pyplot as plt 
import numpy as np 
data = [1.1,4.2,5.3,8.6,10.0,1.2,41.4,23.2] 
dat2 = [1,1,1,1,2,2,2,2] 
ind = [i for i,v in enumerate(dat2) if v==1] 
bins = np.arange(0,45,5) 
plt.hist(data[ind],bins) 

Точки ошибок в Hist () и говорит: «TypeError: индексы списка должны быть целыми числами, а не списком». Я пробовал ind=map(int,ind) и ind=[int(i) for i in ind] без везения.

Answer 1

data = [ d1 for (d1, d2) in zip(data, dat2) if d2 == 1 ] 
plt.hist(data) 

Это молниеносно data и dat2 вместе, создавая список кортежей [ (1.1, 1), (4.2, 1) ... ]. Затем вы можете использовать понимание списка, чтобы сохранить только те кортежи, вторым элементом которых является 1.

Наконец, если списки должны быть достаточно длинными для того, чтобы память была проблемой, вы можете заменить zip на itertools.izip, который возвращает итератор для зашифрованных списков, а не строит их явно.

Answer 2

Когда you'r делать математику, вы обычно используете ndarray объект в numpy ПАКЕТ, которая специально позволяет такого рода индексации:

data = np.array(data) 
... 
data[ind]