【python】配列への格納を適切に行いたい

Question

配列への格納を適切に行いたいです。

プログラム

########## 関数 ##########
def straged_lists(get_elements):
# 取得したelementをひとつずつ分解する
    for item in get_elements:
        # 空の場合実行しない
        if len(item.text) != 0:
            #print(item.text)
            #output_test = item.text.replace('"', '')
            #writer.writerow([str_days, item.text])
            #writer.writerow([item.text])
            writer.writerow([item.text])
            output_lists.append(item.text)
########## 関数ここまで ##########

# 各要素格納配列用意
output_lists = []

# 風
get_elements = driver.find_elements_by_css_selector('tr.td-windCombined')
# 関数として各処理実行
straged_lists(get_elements)

# print(type(output_lists))

#　stringに変換
output_lists_str = ' '.join(map(str, output_lists))

s = '-'.join([str(i) for i in output_lists])
print("----s-----")
print(s)

print("--- len(output_lists) ---")
print(len(output_lists))

print("--- len(output_lists_str) ---")
print(len(output_lists_str))

print("--- output_lists_str out ---")

num = 0
while num < 10:
     # print('output_lists_str[%d] = ' + s[num] % (num))
     print('output_lists_str No.%d = %s' % (num, s[num]))
     num += 1

上記プログラムの出力

---- output_lists -----
['#\n22\n32\n#\n14\n31\n#\n12\n20\n#\n11\n18\n#\n4\n16\n#\n11\n16\n#\n11\n16\n#\n10\n16\n#\n9\n14\n#\n12\n17\n#\n11\n18\n#\n16\n22\n#\n17\n28\n#\n12\n23\n#\n9\n16\n#\n9\n15\n#\n9\n20\n#\n8\n18\n#\n18\n26\n#\n29\n41\n#\n35\n58\n#\n25\n50\n#\n13\n34\n#\n23\n33\n#\n15\n31\n#\n14\n23\n#\n6\n21\n#\n4\n12\n#\n7\n10\n#\n11\n16\n#\n10\n18\n#\n16\n24\n#\n12\n23\n#\n9\n18\n#\n5\n13\n#\n6\n11\n#\n5\n10\n#\n11\n16\n#\n9\n16\n#\n10\n15\n#\n9\n15\n#\n9\n15\n#\n19\n25\n#\n23\n31\n#\n31\n43\n#\n33\n46\n#\n39\n55\n#\n26\n56']
---- s -----
#
22
32
#
14
31
・
・
・
(略)
・
・
・
#
39
55
#
26
56
--- len(output_lists) ---
1
--- len(output_lists_str) ---
367
--- output_lists_str out ---
output_lists_str No.0 = #
output_lists_str No.1 =

output_lists_str No.2 = 2
output_lists_str No.3 = 2
output_lists_str No.4 =

output_lists_str No.5 = 3
output_lists_str No.6 = 2
output_lists_str No.7 =

output_lists_str No.8 = #
output_lists_str No.9 =

理想の出力

"#"や"(空白)"を除外し、各配列に数字を格納したいです。

--- output_lists_str out ---
output_lists_str No.0 = 22
output_lists_str No.1 = 32 
・
・
・

配列の問題かとは思いますが、何卒よろしくお願い申し上げます。

Answer 1

1. output_listsから#\nを取り除く(replace)
2. \nで分割
3. for文で回す

と変えてみました。

output_lists = ['#\n22\n32\n#\n14\n31\n#\n12\n20\n#\n11\n18\n#\n4\n16\n#\n11\n16\n#\n11\n16\n#\n10\n16\n#\n9\n14\n#\n12\n17\n#\n11\n18\n#\n16\n22\n#\n17\n28\n#\n12\n23\n#\n9\n16\n#\n9\n15\n#\n9\n20\n#\n8\n18\n#\n18\n26\n#\n29\n41\n#\n35\n58\n#\n25\n50\n#\n13\n34\n#\n23\n33\n#\n15\n31\n#\n14\n23\n#\n6\n21\n#\n4\n12\n#\n7\n10\n#\n11\n16\n#\n10\n18\n#\n16\n24\n#\n12\n23\n#\n9\n18\n#\n5\n13\n#\n6\n11\n#\n5\n10\n#\n11\n16\n#\n9\n16\n#\n10\n15\n#\n9\n15\n#\n9\n15\n#\n19\n25\n#\n23\n31\n#\n31\n43\n#\n33\n46\n#\n39\n55\n#\n26\n56']
output_without_sharp = output_lists[0].replace("#\n","")
output_split = list(map(int, output_without_sharp.split("\n")))
for num in range(10):
     print(f'output_lists_str No.{num} = {output_split[num]}')
# output_lists_str No.0 = 22
# output_lists_str No.1 = 32
# output_lists_str No.2 = 14
# output_lists_str No.3 = 31
# output_lists_str No.4 = 12
# output_lists_str No.5 = 20
# output_lists_str No.6 = 11
# output_lists_str No.7 = 18
# output_lists_str No.8 = 4
# output_lists_str No.9 = 16

Comments

[t-_-]

早速のご回答ありがとうございます。
素晴らしいです。
printでnumを出力する部分も非常に簡潔で
勉強になりました。

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

Answer 2

現状のコードで得られた文字列を分解するのではなく、個別に文字列を得る方向の回答です。

質問文のコードでは、driver.find_elements_by_css_selector('tr.td-windCombined')でtr要素を選択していますが、実際にデータが書かれているのは、tr要素の中のtd要素です。得られた結果が、ひとつの長い文字列になっているのは、これが原因です。

driver.find_elements_by_css_selector('tr.td-windCombined td')のように、子孫セレクターを使えば、td要素のリストがget_elementsに得られます。変数名からして、意図しているのはこちらでは?

こうして修正したget_elementsをイテレーターとしてfor文で繰り返し処理すると、それぞれのtd要素が得られ、textによって内容のテキスト部分が得られます。具体的には、#\n22\n32のように、改行(\n)で区切られた3行分のデータです。

この質問には書いてありませんが、質問者の以前の質問から類推すると、2行目の値が「風速」、3行目の値が「瞬間最大風速」ですね。

そこで、これらの値を2つのリストに分けて格納するようにします。得られたテキストをsplitして、2つめと3つめの値をそれぞれリストに(整数化してから)appendすればいいでしょう。

get_elements = driver.find_elements_by_css_selector('tr.td-windCombined td')

wind_speed_list = []
max_wind_speed_list = []
for item in get_elements:
    lst = item.text.split()
    wind_speed_list.append(int(lst[1]))
    max_wind_speed_list.append(int(lst[2]))

print(wind_speed_list)
print(max_wind_speed_list)

[22, 14, 12, 11, 4, 11, 11, 10, 9, 12, 11, 16, 17, 12, 9, 9, 9, 8, 18, 29, 35, 25, 13, 23, 15, 14, 6, 4, 7, 11, 10, 16, 12, 9, 5, 6, 5, 11, 9, 10, 9, 9, 19, 23, 31, 33, 39, 26]
[32, 31, 20, 18, 16, 16, 16, 16, 14, 17, 18, 22, 28, 23, 16, 15, 20, 18, 26, 41, 58, 50, 34, 33, 31, 23, 21, 12, 10, 16, 18, 24, 23, 18, 13, 11, 10, 16, 16, 15, 15, 15, 25, 31, 43, 46, 55, 56]