Define a function that extract a float and do operation

Jason_mils 0
Viewed 32

I have a column with the following type of values:

enter image description here

I have to define a function that convert column Diameter in float and manage that kind of exception. In particular:

  • when is 42x54 then make the operation: sqrt(42^2+54^2)
  • when is Steel then return a NaN value
function pandas python
1 Answers

You can use str.extract with a regex to get the numbers, then square them with pow, sum the columns, and get the square root with numpy.sqrt:

import numpy as np

df['Diameter2'] = np.sqrt(df['Diameter']
                          .str.extract('(\d+)(?:\s*x\s*(\d+))?')
                          .astype(float).pow(2)
                          .sum(axis=1, min_count=1)
                          )

output:

  Diameter  Diameter2
0       44  44.000000
1  42 x 54  68.410526
2    Steel        NaN

regex demo

(\d+)              # capture a number
(?:\s*x\s*(\d+))?  # capture a number (optionally) if preceded by x with optional spaces
mozway0
Related