通过条件从数据帧生成图形/矩阵

pip install pandas matplotlib

import pandas as pd
import matplotlib.pyplot as plt

# 创建一个示例数据帧
data = {'A': [1, 2, 3, 4, 5],
        'B': [10, 20, 30, 40, 50],
        'C': [100, 200, 300, 400, 500]}
df = pd.DataFrame(data)

# 定义条件
condition = df['A'] > 2

# 根据条件筛选数据
filtered_data = df[condition]

# 绘制图形
plt.plot(filtered_data['A'], filtered_data['B'])
plt.xlabel('A')
plt.ylabel('B')
plt.title('图形：A > 2')
plt.show()

import pandas as pd

# 创建一个示例数据帧
data = {'A': ['foo', 'foo', 'bar', 'bar', 'foo', 'foo', 'bar', 'bar'],
        'B': ['one', 'two', 'one', 'two', 'one', 'two', 'one', 'two'],
        'C': [10, 20, 30, 40, 50, 60, 70, 80]}
df = pd.DataFrame(data)

# 定义条件
condition = df['A'] == 'foo'

# 根据条件筛选数据
filtered_data = df[condition]

# 生成矩阵
matrix = filtered_data.pivot_table(index='A', columns='B', values='C', aggfunc='sum')

print(matrix)