Итерация по JSON массиву в Python совпадению массива значений - PullRequest
Новая
       6

Итерация по JSON массиву в Python совпадению массива значений

0 голосов
/ 23 января 2020

Я работаю с songkick api, найденным здесь , и я очень близок к завершению программы, над которой я работаю, чтобы получить информацию о нескольких конкретных c артистах предстоящих шоу. Я ввел массив metro_areas, который хочу отслеживать и выводить ТОЛЬКО показы в ЭТИХ городах и их сопровождающие идентификаторы. Другая информация, которую я собираю - это дата шоу, имя исполнителя, название места проведения. По сути, сейчас моя программа может извлекать каждое шоу из списка artist_ids, который я вводил, перебирая идентификаторы в URL запроса для диапазона дат в моих параметрах. Мой текущий вывод выглядит примерно так:

['Date', 'Artist Name', 'Venue Name', 'City', 'metroArea ID'] 

['FEB - 7', 'Rosalia', 'NOTO', 'Philadelphia, PA, US', 5202]
['FEB - 8', 'Rosalia', 'Audio', 'San Francisco, CA, US', 26330]
['FEB - 8', 'Kid Cudi', 'Shady Park', 'Tempe, AZ, US', 23068]
['FEB - 8', 'Kid Cudi', 'Madison Square Garden', 'New York City, NY, US', 7644]

Я хочу вывод будет выглядеть следующим образом: 

['FEB - 8', 'Rosalia', 'Audio', 'San Francisco, CA, US', 26330]
['FEB - 8', 'Kid Cudi', 'Madison Square Garden', 'New York City, NY, US', 7644]

На основании этого массива я определил в начале моей программы совпадение с metro_areas в массиве объектов songkick. 

metro_areas = [
    ('Los Angeles', '17835'),
    ('San Francisco', '26330'),
    ('New York City', '7644'),
    ('Seattle', '2846'),
    ('Nashville', '11104')
]

Вот массив объектов json, который я извлекаю для каждого artist_id: 

{
  "resultsPage": {
    "results": {
      "event": [
        {
          "id":11129128,
          "type":"Concert",
          "uri":"http://www.songkick.com/concerts/11129128-wild-flag-at-fillmore?utm_source=PARTNER_ID&utm_medium=partner",
          "displayName":"Wild Flag at The Fillmore (April 18, 2012)",
          "start": {
            "time":"20:00:00",
            "date":"2012-04-18",
            "datetime":"2012-04-18T20:00:00-0800"
          },
          "performance": [
            {
              "artist": {
                "id":29835,
                "uri":"http://www.songkick.com/artists/29835-wild-flag?utm_source=PARTNER_ID&utm_medium=partner",
                "displayName":"Wild Flag",
                "identifier": []
              },
              "id":21579303,
              "displayName":"Wild Flag",
              "billingIndex":1,
              "billing":"headline"
            }
          ],
          "location": {
            "city":"San Francisco, CA, US",
            "lng":-122.4332937,
            "lat":37.7842398
          },
          "venue": {
            "id":6239,
            "displayName":"The Fillmore",
            "uri":"http://www.songkick.com/venues/6239-fillmore?utm_source=PARTNER_ID&utm_medium=partner",
            "lng":-122.4332937,
            "lat":37.7842398,
            "metroArea": {
              "id":26330,
              "uri":"http://www.songkick.com/metro-areas/26330-us-sf-bay-area?utm_source=PARTNER_ID&utm_medium=partner",
              "displayName":"SF Bay Area",
              "country": { "displayName":"US" },
              "state": { "displayName":"CA" }
            }
          },
          "status":"ok",
          "popularity":0.012763
        }, ....
      ]
    },
    "totalEntries":24,
    "perPage":50,
    "page":1,
    "status":"ok"
  }
}

Еще немного кода, чтобы увидеть, как я получаю вывод из JSON в запросах песенник. 

metro_areas = [
                    ('Los Angeles','17835'),
                    ('San Francisco', '26330'),
                    ('New York City','7644'),
                    ('Seattle','2846'),
                    ('Nashville','11104')
               ]



# artists we want to track
artist_ids = [
    ('Rosalia', '4610868'), ('EARTHGANG', '5720759'), ('Kid Cudi', '8630279'), ('Kanye West', '5566863'),
    ('Ludacris', '398291'), ('Hayley Williams', '10087966')
]

# Fetch existing events in each metro area
for artist_id in artist_ids:

params = {
        'apikey': 'API_KEY',
        'min_date': '2020-02-01',
        'max_date': '2020-02-08',
        # 'type': 'Concert'
    }

r = requests.get('https://api.songkick.com/api/3.0/artists/' + artist_id[1] + '/calendar.json', params=params)
    response = r.json()

shows = response['resultsPage']['results']

    for show in shows:
        try:
            shows = shows['event']

            formatted_shows = [{
                'artistID': [perf['artist']['id'] for perf in s['performance']],
                'date': s['start']['date'],
                'name': [perf['artist']['displayName'] for perf in s['performance']],
                'metroArea': s['venue']['metroArea']['id'],
                'city': s['location']['city'],
                'venue': s['venue']['displayName']
                }
                for s in shows if len(s['performance']) > 0
            ]
            for sub in formatted_shows:
                if sub['artistID'] == artist_id[1]:
                    sub['name'] = artist_id[0]
new_show = artist_id[1]
                new_show_name = artist_id[0]
                new_date = sub['date']
                new_date_time = new_date = datetime.strptime(new_date, '%Y-%m-%d')
                date_time_fin = new_date_time.strftime('%b - %-d').upper()

                formatted_show_final = [date_time_fin, new_show_name, sub['venue'], sub['city'], sub['metroArea']
                print(formatted_show_final)

Короче говоря, мне нужно найти способ перебирать только каждый из моих перечисленных идентификаторов metro_areas (LA, SF, NY C, Seattle, Nashville) и выводить только те шоу, которые соответствуют с 'metroArea': s['venue']['metroArea']['id'] для каждой итерации запроса.

1 Ответ

1 голос
/ 23 января 2020

Если я хорошо понял вопрос, добавьте внутрь второго для l oop: if sub['metroArea'] in [area[1] for area in metro_areas]:

    for show in shows:
        try:
            shows = shows['event']
            formatted_shows = [{
                'artistID': [perf['artist']['id'] for perf in s['performance']],
                'date': s['start']['date'],
                'name': [perf['artist']['displayName'] for perf in s['performance']],
                'metroArea': s['venue']['metroArea']['id'],
                'city': s['location']['city'],
                'venue': s['venue']['displayName']
                }
                for s in shows if len(s['performance']) > 0
            ]
            for sub in formatted_shows:
                #Modified here to apply str() function to transform #sub['metroArea'] to string instead of int value
                if str(sub['metroArea']) in [area[1] for area in metro_areas]:
                    if sub['artistID'] == artist_id[1]:
                        sub['name'] = artist_id[0]
                    new_show = artist_id[1]
                    new_show_name = artist_id[0]
                    new_date = sub['date']
                    new_date_time = new_date = datetime.strptime(new_date, '%Y-%m-%d')
                    date_time_fin = new_date_time.strftime('%b - %-d').upper()
                    formatted_show_final = [date_time_fin, new_show_name, sub['venue'], sub['city'], sub['metroArea']]
                    print(formatted_show_final)
...