pyjanitor，一个强大的 Python 库！

大家好，我是涛哥，本文内容来自 涛哥聊Python ，转载请标原创。

今天为大家分享一个强大的 Python 库 - pyjanitor

Github地址：https://github.com/pyjanitor-devs/pyjanitor

---

Python pyjanitor库是一个用于数据清洗和预处理的强大工具，提供了丰富的函数和方法，可以帮助用户轻松处理数据，清理数据集中的异常值和缺失值，并进行数据转换和整合。本文将介绍如何安装pyjanitor库、其特性、基本功能、高级功能、实际应用场景，并对其进行总结和分析。

安装

安装pyjanitor库非常简单，可以通过pip工具进行安装：

pip install pyjanitor

安装完成后，即可开始使用pyjanitor库进行数据清洗和预处理。

特性

• 数据清洗：提供丰富的函数和方法，可以帮助用户清洗数据集中的异常值、缺失值和重复值。
• 数据转换：支持数据类型转换、列重命名、列排序等操作，方便用户进行数据转换和整合。
• 数据分析：提供统计分析函数，可以帮助用户快速进行数据分析和可视化。
• 数据导出：支持将清洗后的数据导出为CSV、Excel等格式，方便用户进行后续分析和应用。

基本功能

1. 数据清洗

pyjanitor库提供了多种函数和方法，可以帮助用户清洗数据集中的异常值、缺失值和重复值。

下面是一个简单的数据清洗示例代码：

import pandas as pd
import janitor as jn

# 创建示例数据集
data = {
    'name': ['Alice', 'Bob', 'Charlie', 'David'],
    'age': [25, None, 30, 35],
    'salary': [50000, 60000, 70000, 80000]
}
df = pd.DataFrame(data)

# 清洗数据集
cleaned_df = (
    df
    .remove_empty()
    .remove_columns(['name'])
    .fill_missing(value={'age': 0})
)

print(cleaned_df)

在上述代码中，使用了pyjanitor库的remove_empty、remove_columns和fill_missing等函数进行数据清洗，处理了数据集中的空值和缺失值。

2. 数据转换

pyjanitor库还支持数据类型转换、列重命名、列排序等操作，方便用户进行数据转换和整合。

下面是一个简单的数据转换示例代码：

import pandas as pd
import janitor as jn

# 创建示例数据集
data = {
    'Name': ['Alice', 'Bob', 'Charlie', 'David'],
    'Age': [25, 30, 35, 40],
    'Salary': [50000, 60000, 70000, 80000]
}
df = pd.DataFrame(data)

# 数据转换
transformed_df = (
    df
    .rename_column('Name', 'EmployeeName')
    .transform_column('Age', lambda x: x + 1)
    .select_columns(['EmployeeName', 'Age', 'Salary'])
)

print(transformed_df)

在上述代码中，使用了pyjanitor库的rename_column、transform_column和select_columns等函数进行数据转换，修改了列名并对年龄进行了转换。

高级功能

1. 自定义函数

pyjanitor库支持用户自定义函数，可以根据实际需求编写自定义函数进行数据处理和转换。

下面是一个简单的自定义函数示例代码：

import pandas as pd
import janitor as jn

# 自定义函数：将工资转换为年薪
def calculate_annual_salary(salary):
    return salary * 12

# 创建示例数据集
data = {
    'Name': ['Alice', 'Bob', 'Charlie', 'David'],
    'Salary': [50000, 60000, 70000, 80000]
}
df = pd.DataFrame(data)

# 使用自定义函数进行数据转换
transformed_df = (
    df
    .transform_column('Salary', calculate_annual_salary)
)

print(transformed_df)

在上述代码中，定义了一个自定义函数calculate_annual_salary，用于将工资转换为年薪，然后使用该自定义函数对数据集中的工资列进行转换。

2. 数据分析和可视化

pyjanitor库集成了数据分析和可视化的功能，可以帮助用户快速进行数据分析和可视化。

下面是一个简单的数据分析和可视化示例代码：

import pandas as pd
import janitor as jn
import matplotlib.pyplot as plt

# 创建示例数据集
data = {
    'Name': ['Alice', 'Bob', 'Charlie', 'David'],
    'Age': [25, 30, 35, 40],
    'Salary': [50000, 60000, 70000, 80000]
}
df = pd.DataFrame(data)

# 数据分析和可视化
summary_stats = df.describe()
print(summary_stats)

df.plot(kind='bar', x='Name', y='Salary')
plt.title('Employee Salary')
plt.xlabel('Employee Name')
plt.ylabel('Salary')
plt.show()

在上述代码中，使用了pyjanitor库的describe函数进行数据统计分析，并利用matplotlib库进行数据可视化，绘制了员工工资的柱状图。

实际应用场景

1. 数据清洗和整合

pyjanitor库可以帮助用户轻松进行数据清洗和整合，清理数据集中的异常值、缺失值和重复值，并进行数据类型转换和列重命名等操作。

以下是一个简单的数据清洗和整合示例：

import pandas as pd
import janitor as jn

# 创建示例数据集
data = {
    'name': ['Alice', 'Bob', 'Charlie', 'David'],
    'age': [25, None, 30, 35],
    'salary': [50000, 60000, 70000, 80000]
}
df = pd.DataFrame(data)

# 数据清洗和整合
cleaned_df = (
    df
    .remove_empty()
    .fill_missing(value={'age': 0})
    .rename_column('name', 'employee_name')
)

print(cleaned_df)

在上述代码中，使用了pyjanitor库的remove_empty、fill_missing和rename_column等函数进行数据清洗和整合，处理了数据集中的空值和缺失值，并修改了列名。

2. 数据分析和可视化

pyjanitor库也可以帮助用户进行数据分析和可视化，提供了各种统计分析和数据可视化的函数。

以下是一个简单的数据分析和可视化示例：

import pandas as pd
import janitor as jn
import matplotlib.pyplot as plt

# 创建示例数据集
data = {
    'name': ['Alice', 'Bob', 'Charlie', 'David'],
    'age': [25, 30, 35, 40],
    'salary': [50000, 60000, 70000, 80000]
}
df = pd.DataFrame(data)

# 数据分析和可视化
summary_stats = df.describe()
print(summary_stats)

df.plot(kind='bar', x='name', y='salary')
plt.title('Employee Salary')
plt.xlabel('Employee Name')
plt.ylabel('Salary')
plt.show()

在上述代码中，使用了pyjanitor库的describe函数进行数据统计分析，并利用matplotlib库进行数据可视化，绘制了员工工资的柱状图。

3. 数据预处理

除了数据清洗和分析，pyjanitor库还可以进行数据预处理，包括特征工程、数据转换和归一化等操作。

以下是一个简单的数据预处理示例：

import pandas as pd
import janitor as jn
from sklearn.preprocessing import MinMaxScaler

# 创建示例数据集
data = {
    'name': ['Alice', 'Bob', 'Charlie', 'David'],
    'age': [25, 30, 35, 40],
    'salary': [50000, 60000, 70000, 80000]
}
df = pd.DataFrame(data)

# 数据预处理
scaler = MinMaxScaler()
scaled_data = scaler.fit_transform(df[['age', 'salary']])
df[['age', 'salary']] = scaled_data

print(df)

在上述代码中，使用了pyjanitor库进行数据预处理，利用MinMaxScaler对年龄和工资进行归一化处理。

总结

Python pyjanitor库是一个功能强大的数据清洗和预处理工具，具有丰富的函数和方法，可以帮助用户轻松处理数据，清理数据集中的异常值和缺失值，并进行数据转换和整合。通过本文的介绍，读者可以更加深入地了解pyjanitor库的特性、基本功能、高级功能和实际应用场景，掌握如何利用pyjanitor库进行数据清洗和预处理，并在实际应用中取得更好的效果。