cosmos_query_afg.py

# Initialize list
dflist = []
# Connection link
collection_link = database_link + '/colls/' + 'HDIcontainer'
# Write out query
query = 'SELECT * FROM c where c.country="Afghanistan" and c.level="National"'

# For-loop to retrieve individual json records from Cosmos DB 
# that satisfy our query
for item in client.QueryItems(collection_link,
                              query,
                              {'enableCrossPartitionQuery': True}
                              ):
    # Append each item as a dictionary to list
    dflist.append(dict(item))
    
# Convert list to pandas DataFrame
df = pd.DataFrame(dflist)
df.head()