use string content as variable - python

Question

I am using a package that has operations inside the class (? not sure what either is really), and normally the data is called this way data[package.operation]. Since I have to do multiple operations thought of shortening it and do the following

list =["o1", "o2", "o3", "o4", "o5", "o6"]
for i in list:
     print data[package.i]

but since it's considering i as a string it doesnt do the operation, and if I take away the string then it is an undefined variable. Is there a way to go around this? Or will I just have to write it the long way?.

In particular I am using pymatgen, its package Orbital and with the .operation I want to call specific suborbitals. A real example of how it would be used is data[0][Orbital.s], the first [0] denotes the element in question for which to get the orbitals s (that's why I omitted it in the code above).

store the operations as the object not as a string. then i will be the object instead of a string. — MooingRawr
– MooingRawr, Commented Feb 28, 2017 at 17:42
Use getattr(package, i). With getattr you can retrieve attributes of objects via their names as strings. — a_guest
– a_guest, Commented Feb 28, 2017 at 17:43

a_guest · Accepted Answer · 2018-10-29 18:13:47Z

2

You can use getattr in order to dynamically select attributes from objects (the Orbital package in your case; for example getattr(Orbital, 's')).

So your loop would be rewritten to:

for op in ['o1', 'o2', 'o3', 'o4', 'o5', 'o6']:
    print(data[getattr(package, op)])

answered Oct 29, 2018 at 18:13

a_guest

36.6k15 gold badges75 silver badges137 bronze badges

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

Collectives™ on Stack Overflow

use string content as variable - python

1 Answer 1

Comments

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Related