Python numpy ndarray 从文本中跳过行

Question

基于this answer，我使用changethis方法

import numpy as np
import os

def changethis(pos):
    appex = sfile[pos[1]-1][:pos[2]] + '*' + file[pos[1]-1][pos[2]+len(pos[0]):]
    file[pos[1]-1] = appex

pos = ('stack', 3, 16)
sfile = np.genfromtxt('in.cpp',dtype='str',delimiter=os.linesep)
changethis(pos)
print(file)

in.cpp 是一个源文件，其中包含以下内容：

/* Multi-line 
comment
*/

#include <iostream>
#include <fstream>

using namespace std;

int main (int argc, char *argv[]) {
  int linecount = 0;
  double array[1000], sum=0, median=0, add=0;
  string filename;
  if (argc <= 1)
      {
          cout << "Error" << endl;
          return 0;
      }

我得到了输出：

['using namespace std;' 'int main (int argc, char *argv[]) {'
 'int linecount = *' 'double array[1000], sum=0, median=0, add=0;'
 'string filename;' 'if (argc <= 1)' '{' 'cout << "Error" << endl;'
 'return 0;' '}']

请注意，ndarray 中缺少 多行注释、include 语句 和 empty-lines 行.

我不明白为什么会发生这种情况，因为 delimiter 被设置为考虑每个换行符。

关于如何输出的任何建议：

['/* Multi-line' 'comment' '*/' '' '#include <iostream>',
 '' '#include <fstream>' '' 'using namespace std;'
 '' 'int main (int argc, char *argv[]) {'
 'int linecount = *' 'double array[1000], sum=0, median=0, add=0;'
 'string filename;' 'if (argc <= 1)' '{' 'cout << "Error" << endl;'
 'return 0;' '}']

Answer 1

再次抱歉使用genfromtxt，没有理解您的意图，只是试图为问题提供可能的解决方案。作为该特定解决方案的后续行动（已提供其他解决方案），您可以这样做：

import numpy as np
import os

def changethis(pos):
    # Notice file is in global scope
    appex = file[pos[1]-1][:pos[2]] + '*' + file[pos[1]-1][pos[2]+len(pos[0]):]
    file[pos[1]-1] = appex

pos = ('stack', 3, 16)
file = np.array([i for i in open('in.txt','r')]) # instead of genfromtext.
changethis(pos)
print(file)

，导致：

['/* Multi-line \n' 'comment\n' '*/\n*' '\n' '#include <iostream>\n'
 '#include <fstream>\n' '\n' 'using namespace std;\n' '\n'
 'int main (int argc, char *argv[]) {\n' '  int linecount = 0;\n'
 '  double array[1000], sum=0, median=0, add=0;\n' '  string filename;\n'
 '  if (argc <= 1)\n' '      {\n' '          cout << "Error" << endl;\n'
 '          return 0;\n' '      }']

编辑：另一个用户提到的另一个相关点是我用于文件的范围。我并不是要告诉你在全局范围内做事，我的意思是解释该函数正在工作，因为文件在全局范围内。在任何情况下，您都可以创建一个函数来保存范围：

import numpy as np
import os

def changeallthese(poslist,path):
    def changethis(pos):
        appex = file[pos[1]-1][:pos[2]-1] + '*' + file[pos[1]-1][pos[2]-1+len(pos[0]):]
        file[pos[1]-1] = appex
    file = np.array([str(i) for i in open(path,'r')])
    for i in poslist:
        changethis(i)
    return file

poslist = [('stack', 3, 16),('stack', 18, 1),('/* Multi-line', 1, 1)]
file =   changeallthese(poslist,'in.txt')
print(file)

，结果为：

['* \n' 'comment\n' '*/\n*' '\n' '#include <iostream>\n'
 '#include <fstream>\n' '\n' 'using namespace std;\n' '\n'
 'int main (int argc, char *argv[]) {\n' '  int linecount = 0;\n'
 '  double array[1000], sum=0, median=0, add=0;\n' '  string filename;\n'
 '  if (argc <= 1)\n' '      {\n' '          cout << "Error" << endl;\n'
 '          return 0;\n' '* }']

要将数组写入文件，您可以使用 Python 中的普通文件写入系统：

fid = open('out.txt','w')
fid.writelines(file)
fid.close()

，或者使用 numpy 中的函数（但我不确定它是否会添加更多的端线或不那么小心）：

np.savetxt('out.txt',file,fmt='%s')

Answer 2

如果您想要代表文件行的字符串列表，请打开文件并使用readlines()：

with open('in.cpp') as f:
    lines = f.readlines()

# Have changethis take the list of lines as an argument
changethis(lines, pos)

不要使用np.genfromtxt；这是一个表格数据解析器，具有您不想要的各种行为，例如将# 视为行注释标记。

根据您打算如何处理此列表，您甚至可以避免需要明确的行列表。此外，file 是变量名的错误选择（它隐藏了内置的file），changethis 应该真正将列表作为参数而不是全局变量。总的来说，您之前得到的答案非常糟糕。

Answer 3

如果文件不是太大：

import numpy as np
import os

def changethis(linelist,pos):
    appex = linelist[pos[2]-1][:pos[3]] + pos[1] + linelist[pos[2]-1][pos[3]+len(pos[0]):]
    linelist[pos[2]-1] = appex

pos = ('Multi','Three', 1, 3)

with open('in.cpp','r')  as f:
    lines=f.readlines()
    changethis(lines,pos)
print(''.join(lines))

readlines 将您的文件转换为行列表（内存效率低且速度慢，但可以完成工作。如果少于 1k 行应该没问题）。

除了pos 之外，该函数还将行列表作为输入。我还修改了函数，将pos[0] 替换为pos[1]，而不是pos[2] 行和字符pos[3] 之后的*。

我得到这个作为输出：

/* Three-line 
comment
*/

#include <iostream>
#include <fstream>

using namespace std;

int main (int argc, char *argv[]) {
  int linecount = 0;
  double array[1000], sum=0, median=0, add=0;
  string filename;
  if (argc <= 1)
      {
          cout << "Error" << endl;
          return 0;
      }