3

たとえば、同じ長さの 2 つのリストがあります。

a = [[1,2],[2,2],[3,3],[4,2],[5,6]]
b = [1,2,2,3,1]

みたいな機能が欲しい

func(a,b,1) = [[1,2],[5,6]]
func(a,b,2) = [[2,2],[3,3]]
func(b,b,2) = [2,2]

関数が行うことは、a の要素のリストを返すことです。リスト b の同じインデックスの対応する要素は、3 番目の引数に等しくなります。

Matlab では、a(b==1)、a(b==2)、b(b==2) のような簡単なことをします。Pythonでこれを達成する最も効率的な方法は何ですか?

4

4 に答える 4

11

非常にMatlab に似た機能が必要な場合は、numpy を使用できます。

>>> import numpy
>>> a = [[1,2],[2,2],[3,3],[4,2],[5,6]]
>>> b = [1,2,2,3,1]
>>> a = numpy.array(a)
>>> b = numpy.array(b)
>>> a[b==1]
array([[1, 2],
       [5, 6]])
>>> a[b==2]
array([[2, 2],
       [3, 3]])
>>> b[b==2]
array([2, 2])

それができない場合は、おそらく単純にリスト内包表記を使用します。

>>> [i for i,j in zip(a,b) if j == 1]
[[1, 2], [5, 6]]
>>> [i for i,j in zip(a,b) if j == 2]
[[2, 2], [3, 3]]
>>> [i for i,j in zip(b,b) if j == 2]
[2, 2]

これを関数でラップするのは簡単です:

>>> def func(a,b,x):
...     return [i for i,j in zip(a,b) if j == x]
... 
>>> func(a,b,2)
[[2, 2], [3, 3]]
>>>
于 2012-06-21T20:32:18.210 に答える
6

これを行うには、ジェネレーター式itertools.compress()と一緒に使用します。

def func(a, b, c):
    return itertools.compress(a, (x == c for x in b))

これはジェネレーターを返すことに注意してください。リストが必要な場合は、list()呼び出しでラップします。

>>> import itertools
>>> def func(a, b, c):
...     return list(itertools.compress(a, (x == c for x in b)))
... 
>>> a = [[1,2],[2,2],[3,3],[4,2],[5,6]]
>>> b = [1,2,2,3,1]
>>> func(a, b, 1)
[[1, 2], [5, 6]]
>>> func(a, b, 2)
[[2, 2], [3, 3]]
>>> func(b, b, 2)
[2, 2]

また、これは素晴らしく高速である必要があることも注目に値します-itertools高速なモジュールになるように設計されています.

于 2012-06-21T20:35:21.513 に答える
4

bこれは最初にターゲット値 ( )に等しいインデックスを見つけ[i for i, x in enumerate(b) if x==c]、次にそれらのインデックスを使用して必要な要素を取得しますa。

def func(a,b,c):
    return [a[j] for j in [i for i, x in enumerate(b) if x==c]]

>>> func(a,b,1)
[[1, 2], [5, 6]]
>>> func(a,b,2)
[[2, 2], [3, 3]]
>>> func(b,b,2)
[2, 2]
于 2012-06-21T20:31:14.877 に答える
1 
def func(a,b,n):
   return [a[i] for i,x in enumerate(b) if x==n]

a = [[1,2],[2,2],[3,3],[4,2],[5,6]]
b = [1,2,2,3,1]
print(func(a,b,1)) #[[1, 2], [5, 6]]
print(func(a,b,2)) #[[2, 2], [3, 3]]
print(func(b,b,2)) #[2, 2]
于 2012-06-21T20:53:27.410 に答える