Convert Dictionary Values into Array in Python

Question

I know this question is simple but I couldn't get it.

I have a dataframe which is called df_check like the one below

I created a dictionary using the code block below

userdic = df_check.groupby('userid')['poiid'].agg(set).to_dict()

userdict output looks like below

print(userdic)

{1337: {9943, 9996, 10202, 10616}, 1339: {10202, 10725,
11601}, 5: {9177, 9489, 10190, 10349}, 54: {123, 4, ...

My goal is to get the values of each key but I have to take them as arrays because I will be working in for loop.

My Solution to get the first key's values.

dict2list = list(userdic.values())[0]
list2array= np.array(dict2list )

When I print list2array, I get the result as below.

My expectation like this

[360448, 486401, 190984, 941599, ... ]

but output like this

array({360448, 486401, 190984, 941599, ...},dtype=object)

you can just do list(list(userdic.values())[0]), right?

Vinay G
– Vinay G

2021-01-24 15:31:47 +00:00
Commented Jan 24, 2021 at 15:31 — Vinay G
– Vinay G, Commented Jan 24, 2021 at 15:31
set is useful for removing duplicates in the dataset.

ListenSoftware Louise Ai Agent
– ListenSoftware Louise Ai Agent

2021-01-26 18:07:10 +00:00
Commented Jan 26, 2021 at 18:07 — ListenSoftware Louise Ai Agent
– ListenSoftware Louise Ai Agent, Commented Jan 26, 2021 at 18:07

Lukas Schmid · Accepted Answer · 2021-01-24 15:20:17Z

1

Since you aggregated into a set, your numpy array isn't actually an array. To get the list back you can do list(list2array.item()) and index on that

If you change your list(userdic.values())[0] to list(userdic.values()[0]), your initial list will actually be a list instead of a set and the array will get initiated properly

edited Jan 24, 2021 at 15:20

answered Jan 24, 2021 at 15:03

Lukas Schmid

1,9561 gold badge8 silver badges19 bronze badges

Sign up to request clarification or add additional context in comments.

5 Comments

drorhun Over a year ago

When I do print(list2array[0]) I want to see output is 360448 (please see question), I am getting "IndexError: too many indices for array"

drorhun Over a year ago

The other try, when I do print(dict2list[0]) I want to see the output is again 360448 (please see question), I am getting TypeError: 'set' object does not support indexing

Lukas Schmid Over a year ago

I see. Since you aggregated into a set, your numpy array isn't actually an array. To get the list back you can do list(list2array.item()) and index on that

Lukas Schmid Over a year ago

If you change your list(userdic.values())[0] to list(userdic.values()[0]), both your list and your array should work better

Lukas Schmid Over a year ago

changed my answer to reflect the new information

Collectives™ on Stack Overflow

Convert Dictionary Values into Array in Python

1 Answer 1

5 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

5 Comments

Your Answer

Sign up or log in

Post as a guest

Related