python - 在python中读取和操作文件

Question

def ints(filename):
    a = []
    f = open(filename, "r")
    lines = f.readlines()
    f.close()
    for line in lines:
        numbers = line.split()
        for number in numbers:            
            a.append(int(number))
    return a

到目前为止，这是我的功能，我希望能够读取包含整数和字符（如“x”和“b”等）的文件，并返回仅包含整数的列表。目前该函数只能处理包含整数的文件。

如何修改它以排除字符或字母？

Answer 1

这是我对您的代码的编辑，它按您的预期执行。

  def ints(filename):
        """A function that takes the filename of a file as an input arguement, computs and returns a list of
        integers of all the numbers in the file."""
        a = []
        f = open(filename, "r")
        lines = f.readlines()
        f.close()
        for line in lines:
            for character in line:
                try:
                    a.append(int(character))
                except ValueError:
                    pass
        return a

Answer 2

regex在这里可能会有所帮助：

一个简单的例子：

In [22]: import re

In [23]: strs="121 some 34useless text 56"

In [24]: map(int,re.findall("\d+",strs))
Out[24]: [121, 34, 56]

# or this If you want the individual digits:

In [40]: map(int,re.findall("\d",strs))
Out[40]: [1, 2, 1, 3, 4, 5, 6]

对于您的代码，这应该有效：

for line in lines:
    numbers = map(int,re.findall("\d+",line))
    a.extend(numbers)

Answer 3

我只是简单地测试了字符是否为数字：

sample_string = "Test4. 2325This string3"
a_list = []
for x in sample_string:
    if x.isdigit():
        a_list.append(x)

Answer 4

for number in numbers:
    try:
        a.append(int(number))
    except ValueError:
        pass

Answer 5

尝试/捕捉可能会有所帮助：

for thing in line.split():
    i_thing = None
    try:
        i_thing = int(thing)
    except ValueError:
        pass

    s_thing = None
    try:
        s_thing = str(thing)
    except:
        raise Exception("OH NOES!")

这很丑陋，但我还没有找到更好的方法来做你想做的事情。

Answer 6

使用更现代的 Python 习语：

def ints(filename):
    with open(filename, "r") as f:
        for line in f:
            for number in line.split():
                try:
                    yield int(number)
                except ValueError:
                    pass


a = list(ints("testdata.txt"))
print(a)

基本上，尝试转换为 int，如果字符串不是十进制数，则会引发 ValueError。抓住它并忽略它并继续。

Answer 7

如何使用string.translate等将所有非数字替换为空格，然后利用 split() 和 map() 的强大功能。

当然它有点晦涩，我的默认反应是只使用re模块，因为总的来说，你可以用正则表达式做更多的事情，他们值得努力学习。

In [119]: import string
In [120]: allchars = string.maketrans('', '')
In [121]: delchars = allchars.translate(allchars, "0123456789")
In [122]: emptychars = string.maketrans(delchars, ' ' * len(delchars))
In [123]: "Welcome home 1234 56 ol".translate(emptychars)
Out[123]: '             1234 56   '
In [124]: "Welcome home 1234 56 ol".translate(emptychars).split()
Out[124]: ['1234', '56']
In [125]: map(int, "Welcome home 1234 56 ol".translate(emptychars).split())
Out[125]: [1234, 56]

Answer 8

这是未经测试的 sudo 代码，但应该是有效的。另外，我能给出的最佳建议是观看 David Beazley 的系统程序员生成器技巧和掌握 Python 3 I/O 讲座。它们对学习 python 有很大帮助。

这只是一个简单的生成器，用于获取文件的每一行并随后关闭文件。

def getLine(fileName):
    file = open(fileName, "r")

    for line in file.readLines():
        yield line

    file.close()


def getNumbers(line):
// I'm lazy and stole this one from Keith
    for number in line.split():
        try:
            yield int(number)
        except ValueError:
            pass

def generatorChain(fileName):
    // I'm a little iffy on the syntax here, but shouldn't be to hard with a bit of googling
    fileGen = getLine(fileName);
    yield getNumber( fileGen.next() ).next()

def listCompressionMagic():
    return [x for x in generatorChain("foo.txt") ]