从水平行列表中获取部分的一种短（er）方法，用于数独游戏

Question

我现在有密码：

# gets all the horizontal rows from file
rows = [line.strip('\n') for line in open("F:/sudoku practice.txt",'r')]
# gets all the vertical columns from horizontal rows
columns = [(''.join(list(line[i]for line in rows)))for i in range(8)]
import itertools
sections = [[],[],[],[],[],[],[],[],[]]
for line in rows[:3]:
    sections[0].append([(''.join(line[:3]))])
    sections[1].append([(''.join(line[3:6]))])
    sections[2].append([(''.join(line[6:9]))])
for line in rows[3:6]:
    sections[3].append([(''.join(line[:3]))])
    sections[4].append([(''.join(line[3:6]))])
    sections[5].append([(''.join(line[6:9]))])
for line in rows[6:9]:
    sections[6].append([(''.join(line[:3]))])
    sections[7].append([(''.join(line[3:6]))])
    sections[8].append([(''.join(line[6:9]))])
# flattening the list of lists of lists into a list
for i in range(9):
    sections[i] = ''.join(list(itertools.chain.from_iterable(sections[i])))

print(sections)

这将返回：

[' 2 457689', '456 8 237', '789236 4 ', '  5274396', '362 9 574', '9746538  ', ' 4 761938', '618 4 725', '397528 6 ']

列表中的每一项都代表了数独游戏的一部分，这正是我想要的，但是我希望得到所有部分的方法要短得多，我得到了一行和一列，如果可能的话，我想要类似的东西

Answer 1

因为在数独游戏中，您需要大量访问子数组和对角线等，因此Numpy将为您节省大量精力：

import numpy as np

a = np.random.randint(1, 9, (9,9))

print a

[[2 6 3 2 3 4 2 6 5]
 [5 3 7 4 5 2 3 4 7]
 [4 3 1 8 2 7 4 4 2]
 [4 6 5 5 3 6 2 6 8]
 [3 8 3 5 5 4 5 7 3]
 [4 6 2 3 1 6 1 4 2]
 [7 2 6 7 8 3 6 6 3]
 [8 1 7 7 5 7 5 2 1]
 [7 5 3 3 6 1 3 4 2]]

# middle section
print a[3:6,3:6]

[[5 3 6]
 [5 5 4]
 [3 1 6]]

# lower middle diagonals
print a[6:, 3:6].diagonal()

[7 5 1]

# lower middle upward diagonals
print a[6:, 3:6][::-1].diagonal()

[3 5 3]