64 lines
2.3 KiB
Python
64 lines
2.3 KiB
Python
import pandas as pd
|
||
import numpy as np
|
||
|
||
class FixedLengthQueueDataFrame(pd.DataFrame):
|
||
"""一个固定长度的队列式DataFrame"""
|
||
|
||
def __init__(self, data=None, max_size=100, **kwargs):
|
||
if data is not None and len(data) > max_size:
|
||
# 如果初始数据就超过最大长度，只保留最后max_size行
|
||
data = data[-max_size:]
|
||
super().__init__(data=data, **kwargs)
|
||
self._max_size = max_size
|
||
|
||
# 如果初始化后长度仍超过max_size，进行截断
|
||
if len(self) > self._max_size:
|
||
self._truncate()
|
||
|
||
def enqueue(self, item):
|
||
"""入队：添加一行新数据"""
|
||
# 将新数据（字典）转换为DataFrame，确保列对齐
|
||
new_row = pd.DataFrame([item], columns=self.columns)
|
||
# 连接DataFrame
|
||
updated_df = pd.concat([self, new_row], ignore_index=True)
|
||
|
||
# 检查长度并截断
|
||
if len(updated_df) > self._max_size:
|
||
updated_df = updated_df.iloc[-self._max_size:].reset_index(drop=True)
|
||
|
||
# 直接替换当前DataFrame的数据
|
||
self._replace_data(updated_df)
|
||
|
||
def _truncate(self):
|
||
"""如果超过最大长度，截断DataFrame（保留最后max_size行）"""
|
||
if len(self) > self._max_size:
|
||
# 获取最后max_size行
|
||
truncated_data = self.iloc[-self._max_size:].copy().reset_index(drop=True)
|
||
self._replace_data(truncated_data)
|
||
|
||
def _replace_data(self, new_data):
|
||
"""替换当前DataFrame的所有数据"""
|
||
# 清空当前数据
|
||
self.drop(self.index, inplace=True)
|
||
# 添加新数据
|
||
for idx, row in new_data.iterrows():
|
||
self.loc[idx] = row
|
||
|
||
if __name__ == "__main__":
|
||
# 使用示例
|
||
# 初始化一个固定长度为5的队列DataFrame
|
||
queue_df = FixedLengthQueueDataFrame(columns=['温度', '湿度'], max_size=5)
|
||
|
||
# 添加数据
|
||
for i in range(10):
|
||
new_data = {'温度': 20 + i, '湿度': 50 + i}
|
||
queue_df.enqueue(new_data)
|
||
|
||
print("队列当前内容：")
|
||
print(queue_df)
|
||
print("\n队列大小：", len(queue_df))
|
||
|
||
# 直接使用所有Pandas统计函数
|
||
print("\n描述性统计:")
|
||
print(queue_df.describe())
|
||
print("\n温度均值:", queue_df['温度'].mean()) |