better indexing

jamesmkrieger · Nov 20, 2024 · 62a753f · 62a753f
1 parent 692c82b
commit 62a753f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/prody/proteins/interactions.py b/prody/proteins/interactions.py
@@ -228,7 +228,7 @@ def get_energy(pair, source):
     lookup = pair[0]+pair[1]
 
     try:
-        data_results = data[np.where(np.array(aa_pairs)==lookup)[0]][0][2:][np.where(np.array(sources)==source)][0]
+        data_results = data[np.nonzero(np.array(aa_pairs)==lookup)[0]][0][2:][sources.index(source)][0]
     except TypeError:
         raise TypeError('Please replace non-standard names of residues with standard names.')