pythonにおけるマップ処理中のエラー対応のやり方

Question

実現したいこと

pythonで、マップ処理中に例外が発生してしまうのですが、その時の対処方法を知りたいです。

前提

やりたいこととしては、CSVの中にyoutubeのアカウント一覧が入っており、それに紐づくデータを出そうとしています。

youtubeのアカウント名に問題があるとき（URLで入ってしまっているなど）、youtube_channel_detailが動かなくなると思うのですが、その場合には無視して処理を続けたいです。

（python初心者のため、コードの汚い部分はご容赦ください）

発生している問題・エラーメッセージ

Traceback (most recent call last):
  File "...teratail.py", line 24, in <module>
    y3 = list(map(youtube_channel_detail, accounts))
  File "...teratail.py", line 15, in youtube_channel_detail
    return search_response['items'][0]
KeyError: 'items'

該当のソースコード

from apiclient.discovery import build

import pandas as pd

import csv

def youtube_channel_detail(channel_id):
    api_service_name = 'youtube'
    api_version = 'v3'
    youtube = build(api_service_name, api_version, developerKey='API_KEY') #APIキーを入れています
    search_response = youtube.channels().list(
        part='snippet,statistics',
        id=channel_id,
    ).execute()
    return search_response['items'][0]

df = pd.read_csv('youtube_2.csv', skipinitialspace=True)


accounts = df.youtube.tolist()
print(accounts)
print(map(youtube_channel_detail, accounts))

y = list(map(youtube_channel_detail, accounts))

list1 = [d.get('id') for d in y]
list2 = [d.get('statistics').get('subscriberCount') for d in y]
dict_from_list = dict(zip(list1, list2))

with open("dict.csv", 'w') as file:
    writer = csv.writer(file)
    for k, v in dict_from_list.items():
        writer.writerow([k, v])

Comments

[melian]

return search_response['items'][0]

を

return search_response['items'][0] if ('items' in search_response and search_response['items']) else None

として、map の直後で None を削除してみてはどうでしょうか。(None の削除には filter 関数を利用します)

[cat_march]

コメントありがとうございます。

動くようになりました！！

filter関数も説明していただき、とても分かりやすく、助かりました。
ありがとうございました。

Answer 1

以下のようにresponseが意図したデータを保持しているかを確認し、問題あればNoneなり異常を示す値を返すようにします。
あとは呼出元で異常な要素のみを弾いてやればよいです。

def youtube_channel_detail(channel_id):
    # 略
    if 'items' in search_response and len(search_response['items']) > 0:
        return search_response['items'][0]
    return None # この行はなくてもよい

# 略
y = list(map(youtube_channel_detail, accounts))
y = [e for e in y if e is not None]
# 略

Answer 2

melianさんのコメントで解決しました。

まず関数においてはif文を使ってNoneを返すようにし、mapの直後でfilterを使いました。