按标签分离Dataframe (将dataframe转换为numpy数组)

import pandas as pd

# 创建一个示例Dataframe
data = {'A': [1, 2, 3],
        'B': [4, 5, 6],
        'C': [7, 8, 9]}
df = pd.DataFrame(data)

# 按标签分离Dataframe
label = 'A'  # 指定要分离的标签
selected_columns = [label]  # 包含指定标签的列
remaining_columns = list(set(df.columns) - set(selected_columns))  # 剩余的列

selected_df = df[selected_columns]  # 包含指定标签的Dataframe
remaining_df = df[remaining_columns]  # 剩余的Dataframe

# 将Dataframe转换为numpy数组
selected_array = selected_df.to_numpy()
remaining_array = remaining_df.to_numpy()

print("包含指定标签的Dataframe:")
print(selected_df)
print("转换为numpy数组:")
print(selected_array)

print("剩余的Dataframe:")
print(remaining_df)
print("转换为numpy数组:")
print(remaining_array)