32

次のベクトルが与えられた場合、

a = [1, 2, 3, 4, 5, 6, 7, 8, 9]

次のように、要素が4より大きい「a」のインデックスを特定する必要があります。

idx = [3, 4, 5, 6, 7, 8] 

「idx」の情報は、別のリストXから要素を削除するために使用されます(Xには「a」と同じ数の要素があります)。

del X[idx] #idx is used to delete these elements in X. But so far isn't working.

numpyが役立つかもしれないと聞きました。何か案は?ありがとう!

4

6 に答える 6

39
>>> [i for i,v in enumerate(a) if v > 4]
[4, 5, 6, 7, 8]

enumerate配列内の各アイテムのインデックスと値を返します。したがって、値vが、より大きい場合は、新しい配列4にインデックスを含めます。i

または、リストをその場で変更して、上記のすべての値を除外することもできます4

>>> a[:] = [x for x in a if x<=4]
>>> a 
[1, 2, 3, 4]
于 2012-12-05T06:23:37.610 に答える
19

OK、私はあなたが何を意味するのか理解しています、そしてPythonの1行で十分です:

リスト内包表記を使用する

[ j for (i,j) in zip(a,x) if i >= 4 ]
# a will be the list compare to 4
# x another list with same length

Explanation:
>>> a
[1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> x
['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'j']

Zip関数はタプルのリストを返します

>>> zip(a,x)
[(1, 'a'), (2, 'b'), (3, 'c'), (4, 'd'), (5, 'e'), (6, 'f'), (7, 'g'), (8, 'h'), (9, 'j')]

リスト内包表記は、「in」の後に要素をループし、式を使用して要素を評価し、結果をリストに返すためのショートカットです。また、返す結果の条件を追加することもできます。

>>> [expression(element) for **element** in **list** if condition ]

このコードは、圧縮されたすべてのペアを返すだけです。

>>> [(i,j) for (i,j) in zip(a,x)]
[(1, 'a'), (2, 'b'), (3, 'c'), (4, 'd'), (5, 'e'), (6, 'f'), (7, 'g'), (8, 'h'), (9, 'j')]

「if」の後にブール式を指定して条件を追加します。

>>> [(i,j) for (i,j) in zip(a,x) if i >= 4]
[(4, 'd'), (5, 'e'), (6, 'f'), (7, 'g'), (8, 'h'), (9, 'j')]

Itertoolsを使用する

>>> [ _ for _ in itertools.compress(d, map(lambda x: x>=4,a)) ]
# a will be the list compare to 4
# d another list with same length

Pythonでitertools.compressを1行で使用して、このタスクを終了します

>>> a = [1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> d = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'j'] # another list with same length
>>> map(lambda x: x>=4, a)  # this will return a boolean list 
[False, False, False, True, True, True, True, True, True]


>>> import itertools
>>> itertools.compress(d, map(lambda x: x>4, a)) # magic here !
<itertools.compress object at 0xa1a764c>     # compress will match pair from list a and the boolean list, if item in boolean list is true, then item in list a will be remain ,else will be dropped
#below single line is enough to solve your problem
>>> [ _ for _ in itertools.compress(d, map(lambda x: x>=4,a)) ] # iterate the result.
['d', 'e', 'f', 'g', 'h', 'j']

itertools.compressの説明、これはあなたの理解のために明らかだと思います:

>>> [ _ for _ in itertools.compress([1,2,3,4,5],[False,True,True,False,True]) ]
[2, 3, 5]
于 2012-12-05T15:57:28.303 に答える
8
>>> import numpy as np
>>> a = np.array(range(1,10))
>>> indices = [i for i,v in enumerate(a >= 4) if v]
>>> indices
[3, 4, 5, 6, 7, 8]

>>> mask = a >= 4
>>> mask
array([False, False, False,  True,  True,  True,  True,  True,  True], dtype=boo
l)
>>> a[mask]
array([4, 5, 6, 7, 8, 9])
>>> np.setdiff1d(a,a[mask])
array([1, 2, 3])
于 2012-12-05T06:24:53.213 に答える
8

私の目に最も簡単なのは、numpyを使用することです

X[np.array(a)>4]#X needs to be np.array as well

説明:np.arrayはaを配列に変換します。

np.array(a)> 4は、保持する必要のあるすべての要素を含むブール配列を提供します

また、Xはbool配列によってフィルター処理されるため、aが4より大きい要素のみが選択されます(残りは破棄されます)。

于 2017-03-14T09:11:12.637 に答える
6

私は少し遅れてここに来たと思います(Numpyを使用すると物事が簡単になりました)。

import numpy as np

# Create your array
a = np.arange(1, 10)
# a = array([1, 2, 3, 4, 5, 6, 7, 8, 9])

# Get the indexes/indices of elements greater than 4 
idx = np.where(a > 4)[0]
# idx = array([4, 5, 6, 7, 8])

# Get the elements of the array that are greater than 4
elts = a[a > 4]
# elts = array([5, 6, 7, 8, 9])

# Convert idx(or elts) to a list
idx = list(idx)
#idx = [4, 5, 6, 7, 8]
于 2019-09-23T11:47:15.617 に答える
1

フィルタ組み込み関数の使用は問題ありません

>>>a = [1, 2, 3, 4, 5, 6, 7, 8, 9]
>>>filter(lambda x : x < 4, a)
[1, 2, 3]

説明

filter(FUN、Iterable)

この式は、Iterableからすべての要素を反復し、引数としてFUN関数に提供します。戻り値がTrueの場合、arugmentは内部リストに追加されます。

ラムダx:x> 4

これは、引数を取り、4より大きい場合はそれをテストし、TrueまたはFalse値を返す無名関数を意味します

あなたの解決策

4より大きいすべての要素を削除しようとしている場合は、ブローしてみてください

>>> a = [1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> filter(lambda x: x<4 ,a)
[1, 2, 3]
于 2012-12-05T06:25:00.917 に答える