Activity

1. twmr commented on May 22, 2021

   

   twmr

   on May 22, 2021

   Contributor

   More actions

   I'm using the following function for this use-case (the name of it could be improved, though)

   def quantitylistrescale(seq, unit=None):
       """Convert list of quantities to a rescaled quantity array
   
       Examples
       --------
   
       >>> import quantities as pq
       >>> a = 3 * pq.m; b = 2.5 * pq.mm
       >>> quantitylistrescale([a, b], 'cm')
       array([3.0e+02, 2.5e-01]) * cm
   
       If no ``unit`` is given, the elements will be rescaled to the unit of the
       first quantity.
   
       >>> quantitylistrescale([a, b])
       array([3.0e+00, 2.5e-03]) * m
   
       >>> quantitylistrescale([])
       Traceback (most recent call last):
           ...
       StopIteration
       >>> quantitylistrescale([3 * pq.m, 4 * pq.V])
       Traceback (most recent call last):
           ...
       ValueError: Unable to convert between units of "V" and "m"
   
       """
       assert all(isinstance(item, pq.Quantity) for item in seq), (
           f"{seq} contains objects which are not of type pq.Quantity")
   
       if unit is None:
           first = next(iter(seq))
           unit = first.dimensionality.string
   
       rescaled_seq = [item.rescale(unit).magnitude for item in seq]
   
       return pq.Quantity(rescaled_seq, unit)
2. xiki-tempula commented on May 23, 2021

   

   xiki-tempula

   on May 23, 2021

   Last edited by xiki-tempula

   Author

   More actions

   I'm using a similar function (without type checking as I know what is in the array) but I felt that it might be good to have it in the package, like pq.array().