Add ion fields

[Anna Velichko]

Hello everyone.

I just started using trident. I have some troubles with adding ion fields. As I understood, trident uses the metallicity field to create abundances of other elements. The thing is in my hdf5 data file the metallicity array consist of 9 arrays of different elements (2-D array instead of 1-D). It looks like:

metallicity =  

[ [el1  el2  el3  .... el9] 

  [el1  el2  el3  .... el9] 

 .....

  [el1  el2  el3  .... el9] ]

Then, when I do 

trident.add_ion_fields(ds, ions=['H', 'C', 'N', 'O', 'Mg', 'Si II'])

and try to plot the data

C = yt.ProjectionPlot(ds, 'x', ('gas', 'C_p0_number_density'), center=c, width=(1, 'Mpc'))

I get the error message:

ValueError: operands could not be broadcast together with shapes (261933,) (261933,9)

I get the same error when I try to plot metallicity data, because of its shape. Do you know how to deal with this? Is it possible to apply the functions like trident.add_ion_fields or  yt.ProjectionPlot to one column of the 2-D array?

Best wishes,

Anna