我有一个包含数据的输入 csv 文件：
a 15
b 14
c 20
d 45
我想生成一个不同的 csv 文件，该文件将包含来自输入文件的完整数据行，但行应该被打乱。
类似的输出文件可能包含值-
b 14
a 15
c 20
d 45
我试过这段代码：
import random
import sys
op=open('random.csv','w+')
ip=open(sys.argv[1],'r')
data=ip.read()
data1=str(random.choices(data))
op.write(data1)
op.close()
【问题讨论】：
您可以使用 Python random 模块中的 shuffle 函数。像这样：
import random
fid = open("example.txt", "r")
li = fid.readlines()
fid.close()
print(li)
random.shuffle(li)
print(li)
fid = open("shuffled_example.txt", "w")
fid.writelines(li)
fid.close()
打印命令的结果是：
['b 14\n', 'a 15\n', 'c 20\n', 'd 45\n']
['d 45\n', 'a 15\n', 'b 14\n', 'c 20\n']
新文件是这样的：
d 45
a 15
b 14
c 20
只需确保在每条原始行的末尾都有一个换行符。
【讨论】：
data 和data1 并告诉我结果。
随机模块中有一个shuffle 函数。另外，您可以readlines() 以获得列表：
>>> ip=open('random.csv','r')
>>> data=ip.readlines()
>>> data
['a 15\n', 'b 14\n', 'c 20\n', 'd 45\n']
>>> from random import shuffle
>>> shuffle(data)
>>> data
['c 20\n', 'd 45\n', 'a 15\n', 'b 14\n']
如果你有一个标题，只需拆分数据，并打乱行：
>>> ip=open('random.csv','r')
>>> data=ip.readlines()
>>> header, rest=data[0], data[1:]
>>> header
'h1 h2\n'
>>> rest
['a 15\n', 'b 14\n', 'c 20\n', 'd 45\n']
>>> shuffle(rest)
>>> rest
['c 20\n', 'd 45\n', 'a 15\n', 'b 14\n']
>>> [header]+rest
['h1 h2\n', 'c 20\n', 'd 45\n', 'a 15\n', 'b 14\n']
使用 with 语句：
>>> with open('random.csv','r') as ip:
... data=ip.readlines()
...
>>> header, rest=data[0], data[1:]
>>> shuffle(rest)
>>> with open('output.csv','w') as out:
... out.write(''.join([header]+rest))
...
>>>
~$ cat output.csv
h1 h2
d 45
b 14
a 15
c 20
【讨论】：
data1=shuffle(data)。你也应该检查with statement。
header, rest=data[0], data[1:] 可以写成header=data[0] + rest=data[1:]，所以 header 是一个包含数据列表第一个元素的变量：所以它是一个字符串，而 rest 是一个包含所有数据，第一个元素除外。
我认为您应该阅读文件的实际行。
ip.readlines()
并且应该使用random.shuffle() 来换行。
目前，您读取了整个字符串，我认为只能从整个文件中随机获取一个字符。
【讨论】：
使用pandas 的另一个镜头。您可以通过以下方式读取 .csv 文件：
df = pd.read_csv('yourfile.csv', header=None)
然后使用df.sample 洗牌。这将返回数据帧的随机样本，其中行被打乱。使用 frac=1 您将整个集合视为样本：
In [18]: df
Out[18]:
0 1
0 a 15
1 b 14
2 c 20
3 d 45
In [19]: ds = df.sample(frac=1)
In [20]: ds
Out[20]:
0 1
1 b 14
3 d 45
0 a 15
2 c 20
如果您需要再次保存新的洗牌文件，您可以：
ds.to_csv('newfile.csv')
【讨论】：
【讨论】：
如果您的 CSV 包含标题，那么您可以像这样使用 pandas 对其进行随机播放。
df = pd.read_csv(file_name) # avoid header=None.
shuffled_df = df.sample(frac=1)
shuffled_df.to_csv(new_file_name, index=False)
这样您可以避免改组标题并从新 CSV 中删除索引。
【讨论】：
我正在根据@cricket 回答使用此代码：
from random import shuffle
with open('input.csv','r') as f1:
data=f1.readlines()
shuffle(data[0])
with open('output.csv','w') as f2:
f2.write(''.join([data[1:]] + data[0]))
【讨论】：
我是这样走的。
import numpy as np
import pandas as pd
df = pd.read_csv("your_csv_file.csv", header=0)
df.reindex(np.random.permutation(df.index))
【讨论】：