在 CSV 文件/国际象棋数据库创建中创建多行

Question

我正在尝试创建一个数据库来训练基本的机器学习算法。 但是，当我运行代码时，它只创建了两行，但我试图为游戏中的每个 position 创建多行，并在最后对 position 进行鱼分析。 该代码似乎正在进行三个动作，然后写入文件或覆盖每个动作的文件。 我不能说它是哪一个。 进一步阐述，下面的例子是我写的代码的output，其中number = 3：

但是，我正在寻找这样的东西：

0	1	2	3	4	5	6	7	8
磷	磷						p	p
磷			磷				p	p
磷			磷		p			p

这是我的代码：

import chess
import chess.engine
import random
import numpy
import csv
from stockfish import Stockfish

header = [0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,34,35,36,37,38,39,40,41,42,43,44,45,46,47,48,49,50,51,52,53,54,55,56,57,58,59,60,61,62,63]
data = []



def create_file(number):
  board = chess.Board()
  moves = 0
  turns = 0
  black_v = 0
  white_v = 0
  while moves < number:
    if chess.Board.is_game_over(board) == True:
      chess.Board.reset()
    if turns > 200:
      chess.Board.reset()
    for position in header:
      data.append(board.piece_at(position))

    with open('/content/Positions Data/position.csv', 'w', encoding='UTF8', newline='') as f:
        writer = csv.writer(f)
        header.append('sf')
        data.append((stockfish(board, 10))/100)
        writer.writerow(header)
        writer.writerow(data)
        data.clear()
        header.pop()
        turns = turns + 1
        moves = moves + 1
        random_move = random.choice(list(board.legal_moves))
        board.push(random_move)

Answer 1

正如 Camaendir 在评论中指出的那样，您在每次移动迭代时都打开 output 文件。 这将打开“用于创建”的文件，该文件会覆盖任何以前的数据。

此外，通过查看您的代码和您正在尝试做的事情，我看到有两种方法可以实现您想要的，并且您同时尝试这两种方法，这会产生其他问题。

打开 CSV 文件进行写入，并在使用writer.writerow(stockfish(board, 10))/100)处理棋盘/棋步时循环编写棋步：

1. 打开文件进行写入
2. 从文件创建 csv.Writer
3. 写你的 header
4. 循环你的动作
   1. 处理移动/板
   2. 写移动/板
5. 关闭文件

或者，处理附加到data的所有移动/棋盘，然后打开 CSV 文件并写入数据writer.writerows(data) ：

1. 循环你的动作
   1. 处理移动/板
   2. append 表示data
2. 打开文件进行写入
3. 从文件创建 csv.Writer
4. 写你的 header
5. 用data写入所有行