pythonで配列からある要素を取り出したい

Question

pythonのnltkというライブラリであるテキストを形態素解析し、登場頻度順に配列を生成しました。その結果が下記のものです。一行ごとの示す意味は左から単語、品詞、登場数となります。この中から品詞が名詞のものつまり、NNに分類されたもののみを抽出したいのですが、どのようなコードを記載すれば良いでしょうか？
色々と調べているのですが、自分には難しくどなたかご教授いただけないでしょうか。

[(('a', 'DT'), 164),
(('of', 'IN'), 63),
(('on', 'IN'), 21),
(('in', 'IN'), 18),
(('up', 'IN'), 17),
(('at', 'IN'), 15),
(('the', 'DT'), 15),
(('night', 'NN'), 13),
(('close', 'JJ'), 12),
(('water', 'NN'), 10),
(('city', 'NN'), 9),
(('body', 'NN'), 9),
(('with', 'IN'), 9),
(('group', 'NN'), 9),

Answer 1

list comprehension か filter を使えば簡単です。

list comprehension
filter

words = [(('a', 'DT'), 164),
    (('of', 'IN'), 63),
    (('on', 'IN'), 21),
    (('in', 'IN'), 18),
    (('up', 'IN'), 17),
    (('at', 'IN'), 15),
    (('the', 'DT'), 15),
    (('night', 'NN'), 13),
    (('close', 'JJ'), 12),
    (('water', 'NN'), 10),
    (('city', 'NN'), 9),
    (('body', 'NN'), 9),
    (('with', 'IN'), 9),
    (('group', 'NN'), 9)]

# どちらでも可
nouns = [word for word in words if word[0][1] == 'NN']
nouns = list(filter(lambda word: word[0][1] == 'NN', words))
# [(('night', 'NN'), 13),
# (('water', 'NN'), 10),
# (('city', 'NN'), 9),
# (('body', 'NN'), 9),
# (('group', 'NN'), 9)]

Comments

[bullton]

ご回答いただきありがとうございます。
無事解決することができました。

Answer 2

やや複雑なデータ構造になっているので、一時変数に代入して考えると分かりやすいかと思います。

ary = [(('a', 'DT'), 164),
    (('of', 'IN'), 63),
    (('on', 'IN'), 21),
    (('in', 'IN'), 18),
    (('up', 'IN'), 17),
    (('at', 'IN'), 15),
    (('the', 'DT'), 15),
    (('night', 'NN'), 13),
    (('close', 'JJ'), 12),
    (('water', 'NN'), 10),
    (('city', 'NN'), 9),
    (('body', 'NN'), 9),
    (('with', 'IN'), 9),
    (('group', 'NN'), 9)]

item = ary[0]
print(item)  # (('a', 'DT'), 164)

token = item[0] 
print(token) # ('a', 'DT')
cnt = item[1]
print(cnt)   # 164
word = token[0]
print(word)  # a
type = token[1]
print(type)  # DT

ret = []
for item in ary:
    token = item[0]
    type = token[1]
    if type == 'NN':
        ret.append(item)

print(ret)
# [(('night', 'NN'), 13), (('water', 'NN'), 10), (('city', 'NN'), 9), (('body', 'NN'), 9), (('group', 'NN'), 9)]

Comments

[bullton]

ご回答ありがとうございます。
自分の知識では、このような方法を考えることができませんでした。
参考にさせていただきます。