如何使用Python生成3列500万行数据,第一列自动递增,第二列为字符串,第三列为时间戳,保存为csv文件?
时间: 2024-05-14 13:13:06 浏览: 91
可以使用Python中的Pandas库来生成和保存数据。以下是示例代码:
```python
import pandas as pd
import numpy as np
import time
num_rows = 5000000
start_time = int(time.time()) # 获取当前时间戳作为起始时间
# 生成自动递增的整数序列
col1 = pd.Series(np.arange(num_rows) + 1)
# 生成随机的字符串序列,长度为10
col2 = pd.Series([''.join(np.random.choice(list('abcdefghijklmnopqrstuvwxyz'), size=10)) for i in range(num_rows)])
# 生成时间戳序列,每个时间戳为前一个时间戳加上一个随机的偏移量(最大偏移量为3小时)
offsets = np.random.randint(0, 10800, size=num_rows)
timestamps = start_time + np.cumsum(offsets)
col3 = pd.Series(timestamps)
# 创建DataFrame并保存为CSV文件
df = pd.DataFrame({'col1': col1, 'col2': col2, 'col3': col3})
df.to_csv('data.csv', index=False)
```
这将会生成一个名为`data.csv`的文件,包含三列数据:自动递增的整数序列、随机的字符串序列和时间戳序列。
阅读全文