w3resource

Pandas: Extract only number from the specified column of a given DataFrame


27. Extract Only Numbers from Column

Write a Pandas program to extract only number from the specified column of a given DataFrame.

Sample Solution:

Python Code :

import pandas as pd
import re as re
pd.set_option('display.max_columns', 10)
df = pd.DataFrame({
    'company_code': ['c0001','c0002','c0003', 'c0003', 'c0004'],
    'address': ['7277 Surrey Ave.','920 N. Bishop Ave.','9910 Golden Star St.', '25 Dunbar St.', '17 West Livingston Court']
    })
print("Original DataFrame:")
print(df)
def find_number(text):
    num = re.findall(r'[0-9]+',text)
    return " ".join(num)
df['number']=df['address'].apply(lambda x: find_number(x))
print("\Extracting numbers from dataframe columns:")
print(df)

Sample Output:

Original DataFrame:
  company_code                   address
0        c0001          7277 Surrey Ave.
1        c0002        920 N. Bishop Ave.
2        c0003      9910 Golden Star St.
3        c0003             25 Dunbar St.
4        c0004  17 West Livingston Court
\Extracting numbers from dataframe columns:
  company_code                   address number
0        c0001          7277 Surrey Ave.   7277
1        c0002        920 N. Bishop Ave.    920
2        c0003      9910 Golden Star St.   9910
3        c0003             25 Dunbar St.     25
4        c0004  17 West Livingston Court     17

For more Practice: Solve these Related Problems:

  • Write a Pandas program to extract only numeric characters from a string column using regex and then convert them to integers.
  • Write a Pandas program to filter out non-digit characters from a DataFrame column and then output the cleaned numeric values.
  • Write a Pandas program to create a new column that contains only the numbers extracted from a given text column.
  • Write a Pandas program to apply regex extraction to remove all non-numeric characters in a column and then cast the result to a numeric type.

Go to:


Previous: Write a Pandas program to extract word mention someone in tweets using @ from the specified column of a given DataFrame.
Next: Write a Pandas program to extract only phone number from the specified column of a given DataFrame.

Python Code Editor:

Have another way to solve this solution? Contribute your code (and comments) through Disqus.

What is the difficulty level of this exercise?

Test your Programming skills with w3resource's quiz.



Follow us on Facebook and Twitter for latest update.