35 lines
682 B
Python
35 lines
682 B
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
"""
|
||
|
|
Converts a text file containing a list of integers
|
||
|
|
into a format that can be loaded by ivy.
|
||
|
|
"""
|
||
|
|
|
||
|
|
with open("input") as f:
|
||
|
|
lines = f.read().splitlines()
|
||
|
|
|
||
|
|
data = []
|
||
|
|
for line in lines:
|
||
|
|
line = list(line.strip())
|
||
|
|
data.append(line)
|
||
|
|
|
||
|
|
def count(row):
|
||
|
|
n = 0
|
||
|
|
for x in row:
|
||
|
|
if x.isdigit():
|
||
|
|
n += 1
|
||
|
|
else:
|
||
|
|
n += len(x)
|
||
|
|
return n
|
||
|
|
|
||
|
|
cols = max(map(count, data))
|
||
|
|
|
||
|
|
print("input = {} {} rho".format(len(data), cols), end="")
|
||
|
|
for row in data:
|
||
|
|
print("", " ".join(
|
||
|
|
x if x.isdigit() else repr(x)
|
||
|
|
for x in row), end="")
|
||
|
|
n = count(row)
|
||
|
|
if n < cols:
|
||
|
|
print(" 0" * (cols - n), end="")
|