要按布尔分隔符拆分NumPy数组或Pandas数据帧,可以使用以下方法:
对于NumPy数组:
import numpy as np
# 创建一个示例NumPy数组
arr = np.array([1, 0, 1, 1, 0, 0, 1, 0])
# 使用np.split()函数按布尔分隔符拆分数组
split_arr = np.split(arr, np.where(arr)[0])
# 输出拆分后的数组列表
for sub_arr in split_arr:
print(sub_arr)
对于Pandas数据帧:
import pandas as pd
# 创建一个示例Pandas数据帧
df = pd.DataFrame({'A': [1, 2, 3, 4, 5],
'B': [0, 1, 0, 1, 0]})
# 使用pd.groupby()函数按布尔分隔符拆分数据帧
split_df = df.groupby((df['B'] != df['B'].shift()).cumsum())
# 输出拆分后的数据帧
for group, sub_df in split_df:
print(sub_df)
这些代码示例演示了如何使用np.split()函数拆分NumPy数组和如何使用pd.groupby()函数拆分Pandas数据帧。请根据实际情况选择适当的方法来拆分数据。
上一篇:anboxdebian
下一篇:按布尔排序