Extracting dictionary values in for loop to fill list

Question

I have this code:

import numpy as np

result = {}
result['depth'] = [1,1,1,2,2,2]
result['generation'] = [1,1,1,2,2,2]
result['dimension'] = [1,2,3,1,2,3]
result['data'] = [np.array([0,0,0]), np.array([0,0,0]), np.array([0,0,0]), np.array([0,0,0]), np.array([0,0,0]), np.array([0,0,0])]

for v in np.unique(result['depth']):
    temp_v = np.where(result['depth'] ==  v)
    values_v = [result[string][temp_v] for string in result.keys()]
    this_v = dict(zip(result.keys(), values_v))

in which I want to create a new dictcalled 'this_v', with the same keys as the original dict result, but fewer values.

The line:

values_v = [result[string][temp_v] for string in result.keys()]

gives an error

TypeError: list indices must be integers, not tuple

which I don't understand, since I can create ex = result[result.keys()[0]][temp_v] just fine. It just does not let me do this with a for loop so that I can fill the list.

Any idea as to why it does not work?

np.where returns a tuple

DavidG
– DavidG

2018-08-23 11:28:51 +00:00
Commented Aug 23, 2018 at 11:28 — DavidG
– DavidG, Commented Aug 23, 2018 at 11:28

Arghya Saha · Accepted Answer · 2018-08-23 11:37:36Z

2

I'm not sure what you are trying to achieve but I could solve your issue:

np.where is returning a tuple, so to access you need to do give the index temp_v[0]. Also the value of the tuple is an array so to loop over the value you need to run another loop a for a in temp_v[0] which helps you you access the value.

import numpy as np

result = {}
result['depth'] = [1,1,1,2,2,2]
result['generation'] = [1,1,1,2,2,2]
result['dimension'] = [1,2,3,1,2,3]
result['data'] = [np.array([0,0,0]), np.array([0,0,0]), np.array([0,0,0]), np.array([0,0,0]), np.array([0,0,0]), np.array([0,0,0])]

for v in np.unique(result['depth']):
    temp_v = np.where(result['depth'] ==  v)
    values_v = [result[string][a] for a in temp_v[0] for string in result.keys()]
    this_v = dict(zip(result.keys(), values_v))

answered Aug 23, 2018 at 11:37

Arghya Saha

5,7734 gold badges28 silver badges52 bronze badges

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

6 Comments

SuperCiocia Over a year ago

Hey, thanks. I actually found a better way by using a mask. Now I have <code> temp_v = (result['depth'] == v) values_v = [result[string][temp_v] for string in result.keys()]</code> but it still gives an error, "only integer scalar arrays can be converted to a scalar index"

Arghya Saha Over a year ago

try this line values_v = [result[string][a] for a in temp_v[0] for string in result.keys()]

SuperCiocia Over a year ago

That works, but I'm trying to recude the number of for loops. Hence the masking idea

Arghya Saha Over a year ago

could you edit your code in the question, I can help you then

SuperCiocia Over a year ago

I asked it here: stackoverflow.com/questions/51985766/…

|

Collectives™ on Stack Overflow

Extracting dictionary values in for loop to fill list

1 Answer 1

6 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

6 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related