python读取csv文件的前1万行数据
时间: 2023-09-30 21:09:58 浏览: 186
以下是读取CSV文件前1万行数据的Python代码示例:
```python
import csv
# 设置CSV文件路径
csv_file_path = 'example.csv'
# 打开CSV文件并读取前1万行数据
with open(csv_file_path, newline='') as csvfile:
reader = csv.reader(csvfile)
for i, row in enumerate(reader):
if i == 10000:
break
print(row)
```
这段代码使用了Python内置的csv模块来读取CSV文件,并使用enumerate函数来遍历前1万行数据。如果CSV文件有标题行,可以将`if i == 10000:`改为`if i == 10001:`来跳过标题行。
相关问题
python读取csv文件将万转换成纯数字
可以使用 Python 自带的 csv 模块和 re 模块来读取 csv 文件并将万转换为纯数字。以下是一个示例代码:
```python
import csv
import re
with open('data.csv', 'r', encoding='utf-8') as f:
reader = csv.reader(f)
for row in reader:
# 将万转换为纯数字
pattern = re.compile(r'\d+\.?\d*万')
for i in range(len(row)):
if pattern.search(row[i]):
res = round(float(pattern.search(row[i]).group(0)[:-1]) * 10000)
row[i] = str(res)
print(row)
```
其中,`data.csv` 是你要读取的 csv 文件名,可以根据需要修改。这段代码会将 csv 文件中的每一行都转换一遍,将其中的所有万转换为对应的纯数字,并输出转换后的结果。
用python随机读取csv文件5万行并存放在另一个csv文件中
以下是使用Python随机读取CSV文件并将其存储在另一个CSV文件中的示例代码:
```python
import random
import csv
# 打开原始CSV文件
with open('original.csv', 'r') as original_file:
original_csv = csv.reader(original_file)
# 读取第一行(标题行)
header = next(original_csv)
# 创建一个空的列表来存储随机行
random_rows = []
# 随机选择5万行
for i, row in enumerate(original_csv):
if i < 50000:
random_rows.append(row)
else:
# 选择到5万行后,以1/i的概率替换掉之前随机选择的行
replace_index = random.randint(0, i)
if replace_index < 50000:
random_rows[replace_index] = row
# 将随机行写入新的CSV文件
with open('random.csv', 'w', newline='') as random_file:
random_csv = csv.writer(random_file)
# 写入标题行
random_csv.writerow(header)
# 写入随机行
for row in random_rows:
random_csv.writerow(row)
```
这个代码使用Python内置的csv模块来读取和写入CSV文件。它首先打开原始CSV文件,然后使用csv.reader对象逐行读取文件。在读取标题行后,它创建一个空的列表random_rows,用于存储随机行。
然后,它使用一个循环来遍历原始CSV文件中的每一行,直到读取了50000行。对于前50000行,它直接将它们添加到random_rows列表中。对于超过50000行的行,它以1/i的概率替换random_rows列表中的一个随机行,其中i是当前行的索引。
最后,它使用csv.writer对象将随机行写入新的CSV文件。它首先写入标题行,然后写入random_rows列表中的每一行。注意,我们需要将newline参数设置为'',以避免在Windows系统中添加额外的空行。
阅读全文