当前位置: 首页 > 科技迭代

为数据列表中的每个字典添加时间戳

时间:2024-02-20 18:55:32 科技迭代

请确保你的环境中已经安装了 pandas 库。以下是一个 Python 解决方案:

python

import pandas as pd


def add_timestamp(data_list):

 df = pd.DataFrame(data_list)

 # 检查每个字典是否具有相同的键和相同数量的元素

 assert all(len(dict_i) == len(dict_j) for i, dict_i in enumerate(df)) and all(set(dict_i.keys()) == set(dict_j.keys()) for i, dict_i in enumerate(df))

 # 添加时间戳作为索引

 df['stime'] = pd.Timestamp.now()

 return df


# 示例输入

data_list = [{"stock_code": "GOOG", "open": 101.23, "close": 105.32},{"stock_code": "MSFT", "open": 45.34, "close": 47.23}]

# 调用函数并打印结果

df = add_timestamp(data_list)

print(df)

此脚本定义了一个名为 add_timestamp 的函数,该函数接受一个包含多个字典的列表 data_list 作为输入。首先,它将列表转换为 DataFrame 对象,然后检查每个字典是否具有相同的键和相同数量的元素。如果是这样,它将为每个股票代码列表内的数据添加一个时间戳作为索引,并将新的时间戳字段命名为 stime。最后,函数返回带有时间戳的 DataFrame 对象。