#python #data-science #python-3.8
Вопрос:
users = [
{'id': 0, "name": "Hero"},
{'id': 1, "name": "Dunn"},
{'id': 2, "name": "Sue"},
{'id': 3, "name": "Chi"},
{'id': 4, "name": "Thor"},
{'id': 5, "name": "Clive"},
{'id': 6, "name": "Hicks"},
{'id': 7, "name": "Devin"},
{'id': 8, "name": "Kate"},
{'id': 9, "name": "Klein"},
]
# list of users
friendship_pairs = [(0, 1), (0, 2), (1, 2), (1, 3), (2, 3), (3, 4), (4, 5),
(5, 6), (5, 7), (6, 8), (7, 8),
(8, 9)] # list of friendship pairs
# it is easy to make a dictionary list of friendships for the IDs as the friendship_pairs and written in terms of IDs
# what if I wanted this dict to show for example "Hero": ["dunn","Sue"] instead of 0:[1,2]
friendships = {user['id']: [] for user in users}
for i, j in friendship_pairs:
friendships[i].append(j)
friendships[j].append(i)
display(friendships)
# I made a dictonary of intergers:names
user_Ids = list(user['id'] for user in users)
user_Names = list(user['name'] for user in users)
converter = dict(zip(user_IDS, user_Names))
# How can I use this to turn 'friendship_pairs' into something like 'friendship_pairs_names'
# which has the same information has friendship_pairs but uses their names
Я хотел бы иметь возможность принимать входные данные friendship_pairs и возвращать переменную, содержащую ту же информацию, в которой значения идентификаторов пользователей заменяются значениями имен пользователей, чтобы я мог создавать один и тот же словарь, отображаемый в идентификаторах, с именами. Открыт для любых предложений по более быстрым или чистым методам.
Ответ №1:
Напишите словарь, который сопоставляет идентификатор с именем. подобный этому:
names = {user['id']:user['name'] for user in users}
Затем вы можете создать словарь таким же образом, как вы создали его для идентификаторов, но с именами[id] вместо идентификатора, например так:
users =[
{'id':0, "name":"Hero"},
{'id':1, "name":"Dunn"},
{'id':2, "name":"Sue"},
{'id':3, "name":"Chi"},
{'id':4, "name":"Thor"},
{'id':5, "name":"Clive"},
{'id':6, "name":"Hicks"},
{'id':7, "name":"Devin"},
{'id':8, "name":"Kate"},
{'id':9, "name":"Klein"},
]
names = {user['id']:user['name'] for user in users}
friendship_pairs = [(0,1),(0,2),(1,2),(1,3),(2,3),(3,4),(4,5),(5,6),(5,7),(6,8),(7,8),(8,9)]
friendships = {user['name']: [] for user in users}
for i, j in friendship_pairs:
friendships[names[i]].append(names[j])
friendships[names[j]].append(names[i])
print(friendships)
выход:
{
'Hero': ['Dunn', 'Sue'],
'Dunn': ['Hero', 'Sue', 'Chi'],
'Sue': ['Hero', 'Dunn', 'Chi'],
'Chi': ['Dunn', 'Sue', 'Thor'],
'Thor': ['Chi', 'Clive'],
'Clive': ['Thor', 'Hicks', 'Devin'],
'Hicks': ['Clive', 'Kate'],
'Devin': ['Clive', 'Kate'],
'Kate': ['Hicks', 'Devin', 'Klein'],
'Klein': ['Kate']
}
Вы также можете использовать setdefault
as, чтобы не нуждаться в {user['name']: [] for user in users}
понимании словаря, и вместо этого сделать это:
friendships = {}
for i, j in friendship_pairs:
friendships.setdefault(names[i], []).append(names[j])
friendships.setdefault(names[j], []).append(names[i])
Ответ №2:
Вы можете попробовать это
group= [(list(users[x].values())[1],list(users[y].values())[1] )for x,y in friendship_pairs]
Выходы
[('Hero', 'Dunn'),
('Hero', 'Sue'),
('Dunn', 'Sue'),
('Dunn', 'Chi'),
('Sue', 'Chi'),
('Chi', 'Thor'),
('Thor', 'Clive'),
('Clive', 'Hicks'),
('Clive', 'Devin'),
('Hicks', 'Kate'),
('Devin', 'Kate'),
('Kate', 'Klein')]