「指定した要素数」以下の要素を格納したい

Question

###やりたいこと
作成したリストのうち、先頭から３つの要素を別の配列に格納したいです。

top3 = []
for i in range(3):
    if lda_of_test_doc[i] == []:
        continue
    top3.append(lda_of_test_doc[i])

lda_of_test_doc[]の要素数が2以下の場合エラーが出てしまいます。
if文で回避しようとしたのですが、うまくいきませんでした。
また、lda_of_test_doc[]の中身は
[(6, 0.48613339755650264), (0, 0.2976580111959658), (1, 0.19870383657944415)]
このようになっているのですが、
各要素の１つ目？（上の例で言う6,0,1）だけを格納する方法を教えていただきたいです。

Answer 1

先頭から３つの要素を別の配列に格納したい

この条件を素直に考えると

• 元配列が３個未満なら、元配列の個数
• それ以上なら３個

追加すればよいという考えにいたります。

また

各要素の１つ目？（上の例で言う6,0,1）だけを格納する方法

も、元配列の１要素をitemという変数に格納することにより、その要素の１つ目はitem[0]でアクセスできるという考えにいたります。

上記の考えをそのままコードにしたのが以下になります。

lda_of_test_doc = [(6, 0.48613339755650264), (0, 0.2976580111959658), (1, 0.19870383657944415)]

# 最大３個をリストに追加する
append_count = len(lda_of_test_doc)
if append_count > 3:
    append_count = 3

top3 = []
for i in range(append_count):
    item = lda_of_test_doc[i]
    #top3.append(item)
    top3.append(item[0]) # タプルの先頭要素のみを追加

print(top3)

どのようにコードを書いてよいか分からなくなったときは、まず平易な日本語で、やりたいことを箇条書きしながら考えてみるとよいです。

Comments

[aoisj]

うまくいきました。
ありがとうございました。