Python：写长度控制

Posted 2023-03-17

【中文标题】Python：写长度控制

【英文标题】：Python: Write-Length Control

【发布时间】：2021-12-22 13:53:59

【问题描述】：

我正在尝试为稍后要加载到 C++ 中的文本文件创建特定的输出模式。 我用 Python 编写了一个文件，它在一个圆圈中创建随机坐标和运动值。 输出应该有输出：

   Place_1 Place_2 Place_3 Movement_1 Movement_2 Movement_3\n

   Place_1 Place_2 Place_3 Movement_1 Movement_2 Movement_3\n

   Place_1 Place_2 Place_3 Movement_1 Movement_2 Movement_3

代码是使用是

import numpy as np
file = open('log.txt', 'a')


def f(n, center, radius, ecc):
  pos = np.zeros((n,6))
  r = ecc * radius
  for i in range(n):
    while 1:
        x_1 = -1 + 2 * np.random.rand(1)
        x_2 = -1 + 2 * np.random.rand(1)
        if (x_1*x_1 + x_2*x_2)<1 :
            pos[i,0] = center[0] + r * 2 * x_1 * np.sqrt(1 - x_1*x_1 - x_2*x_2)
            pos[i,1] = center[1] + r * 2 * x_2 * np.sqrt(1 - x_1*x_1 - x_2*x_2)
            pos[i,2] = center[2] + r * (1 - 2 * (x_1*x_1 + x_2*x_2))
            pos[i,3] = (-1 + 2 * np.random.rand(1))
            pos[i,4] = (-1 + 2 * np.random.rand(1))
            pos[i,5] = (-1 + 2 * np.random.rand(1))
            break
    string = str(pos[i,:]).strip('[]').rstrip('\n')
    file.write(string)
  return

f(10000, np.array((127,127,127)), 92, 0.9)

file.close()

但是，我创建的日志格式非常糟糕。如何获得所需的格式？

【问题讨论】：

为什么要使用str(np.array) 将列表转换为字符串？可能你最好使用 python 的f-strings

使用 f'pos[i,:]' 我得到输出 `` [ 2.03736833e+02 1.14369585e+02 1.55421313e+02 -1.34950352e-01 -9.72404614e-01 -2.09317199E-01] [1.49366346C + 02 2.06721909C + 02 1.2.06931669S-02 7.564066692S-02 7.564066692S-02 7.564066692S-02 7.564066698C-02 7.564066692S-01 7.5640666998S-01 7.56406688C-01 7.56406688C-01 7.56406688C-01 7.56406688C-01] 3.73819153992 + 02 5.94674798992-017397899898920 + 02 5.94674799992年第7000 + 02 + 01 3.73394744e-01 -2.16969418e-02 -4.58853325e-01][ 9.16926315e+01 1.45733283e+02 1.99514094e+02 3.65330619 ```

遗憾的是，这仍然没有满足必要的格式...

您需要适当地格式化列表，而不仅仅是吐出整个内容；正在研究解决方案。

幸运的是，我已经从 Tim Roberts 那里得到了解决方案，但仍然感谢您！

【参考方案1】：

对于格式问题，您可以将坐标转换为字符（我的意思是将坐标转换为可以打印为 _ 或 / 之类的字符）您也可以在 print(" Hello.") 也可以转到控制台中的另一行执行 print("\n Hello.")。这可能不是您想要的，但我希望这会有所帮助。

【讨论】：

这个答案完全没有帮助。

虽然我很高兴能得到帮助，但我不太明白这里的建议。这将如何解决输出格式问题？可悲的是我不明白你的解决方案的想法。

正如目前所写，您的答案尚不清楚。请edit 添加其他详细信息，以帮助其他人了解这如何解决所提出的问题。你可以找到更多关于如何写好答案的信息in the help center。

以后，如果你必须写“这可能不是你要找的”，那么你不应该发布它。 *** 上的答案应该是高质量的，甚至比问题还要高。

【参考方案2】：

你会在这里遇到很多你不需要的麻烦。这似乎解决了问题，简单。

import numpy as np
file = open('log.txt', 'a')

def f(n, center, radius, ecc):
    r = ecc * radius
    for i in range(n):
        while 1:
            pos = [0]*6
            x_1 = -1 + 2 * np.random.rand(1)[0]
            x_2 = -1 + 2 * np.random.rand(1)[0]
            if (x_1*x_1 + x_2*x_2) < 1:
                pos[0] = center[0] + r * 2 * x_1 * np.sqrt(1 - x_1*x_1 - x_2*x_2)
                pos[1] = center[1] + r * 2 * x_2 * np.sqrt(1 - x_1*x_1 - x_2*x_2)
                pos[2] = center[2] + r * (1 - 2 * (x_1*x_1 + x_2*x_2))
                pos[3] = (-1 + 2 * np.random.rand(1)[0])
                pos[4] = (-1 + 2 * np.random.rand(1)[0])
                pos[5] = (-1 + 2 * np.random.rand(1)[0])
                break
        print(*pos, file=file)

f(10000, np.array((127,127,127)), 92, 0.9)

file.close()

没有numpy的解决方案：

import math
import random
file = open('log.txt', 'a')

def f(n, center, radius, ecc):
    r = ecc * radius
    for _ in range(n):
        pos = [0]*6
        while 1:
            x_1 = 2 * random.random() - 1
            x_2 = 2 * random.random() - 1
            vector = x_1*x_1 + x_2*x_2
            if vector < 1:
                break
        pos[0] = center[0] + r * 2 * x_1 * math.sqrt(1 - vector)
        pos[1] = center[1] + r * 2 * x_2 * math.sqrt(1 - vector)
        pos[2] = center[2] + r * (1 - 2 * vector)
        pos[3] = 2 * random.random() -1
        pos[4] = 2 * random.random() -1
        pos[5] = 2 * random.random() -1
        print(*pos, file=file)

f(10000, (127,127,127), 92, 0.9)

file.close()

【讨论】：

这成功了！非常感谢！

除非你将它用于其他事情，否则你根本不需要 numpy。 random.random 可以很好地达到目的。

@TimRoberts 我认为使用np.savetxt 会更有效；不过，很高兴显示不正确。

唯一的缺点是你必须在写入之前在内存中构造整个数组。我正在写每一行然后丢弃数据。这取决于用例。如果他不需要生成的数组，则无需构建它。

还建议计算一次(x_1*x_1 + x_2*x_2)，并在计算if (value) < 1, pos[0,1,2]时使用它四次，fwiw

【参考方案3】：

使用np.savetxt:

import numpy as np

def f(n, center, radius, ecc):
  pos = np.zeros((n,6))
  r = ecc * radius
  for i in range(n):
    while 1:
        x_1 = -1 + 2 * np.random.rand(1)
        x_2 = -1 + 2 * np.random.rand(1)
        if (x_1*x_1 + x_2*x_2)<1 :
            pos[i,0] = center[0] + r * 2 * x_1 * np.sqrt(1 - x_1*x_1 - x_2*x_2)
            pos[i,1] = center[1] + r * 2 * x_2 * np.sqrt(1 - x_1*x_1 - x_2*x_2)
            pos[i,2] = center[2] + r * (1 - 2 * (x_1*x_1 + x_2*x_2))
            pos[i,3] = (-1 + 2 * np.random.rand(1))
            pos[i,4] = (-1 + 2 * np.random.rand(1))
            pos[i,5] = (-1 + 2 * np.random.rand(1))
            break
  np.savetxt('file.txt',pos,delimiter=';')
  return

f(100, np.array((127,127,127)), 92, 0.9)