python获取代码所在行号，输出到终端或日志文件中

python获取代码所在行号，输出到终端或日志文件中

在编写程序过程中，通常需要获取代码所在的行号，以便进行问题的分析及定位，常用的获取行号的方式有：

• 使用 sys 模块

• 使用inspect模块

• 使用linecache模块

• 使用traceback模块

• 使用enumerate函数

但它们各有特点：

• 使用linecache模块可以很方便地获取指定行的代码内容，但不适用于获取当前行号。

• 使用inspect模块可以获取当前行号以及其他代码信息，但需要在函数或方法中使用。

• 使用traceback模块可以获取当前行号以及其他堆栈信息，但更适合处理异常和追踪代码执行路径。

1、使用 sys 模块

import sys

def PRINT(string):
    print('func:{}(),line:{},'.format(sys._getframe().f_code.co_name, sys._getframe().f_lineno), end="")
    # /print('调用该函数的上级为{}'.format(sys._getframe(1).f_code.co_name))
    print(string)

PRINT("help :%d" % 123654)

2、使用inspect模块

import inspect

def print_line_number():
    frame = inspect.currentframe()
    lineno = frame.f_back.f_lineno
    print(f'Line {lineno}:  Hello, World! ')
    print(f'Line {lineno}: print("Hello, World!")')

print_line_number()

再比如：

import inspect


frame = inspect.currentframe()
lineno = frame.f_back.f_lineno
print(f'Line {lineno}: print("is ok!")')
print(f'Line {inspect.currentframe().f_back.f_lineno}: print("is ok!")')
print(f'Line {inspect.currentframe().f_lineno}: print("is ok!")')

上述代码中，我们使用inspect.currentframe()函数来获取当前代码的堆栈帧对象。然后，通过frame.f_lineno属性来获取行号。最后，我们使用print()函数来打印行号。

需要注意的是，inspect模块是一个非常强大的工具，除了行号，它还可以获取函数、类、方法等的相关信息，可以用于编写更复杂的调试工具。

3、使用linecache模块

Python的linecache模块提供了一些函数用于缓存和提供源代码的行信息。我们可以使用linecache模块来打印特定行号的代码。

import linecache

def print_line_number():
    line = linecache.getline(__file__, 3)
    print(f'Line 3: {line.strip()}')

print_line_number()

4、使用traceback模块

Python的内置模块traceback提供了一种更灵活的方法来获取代码的行号。我们可以利用traceback模块来获取当前代码所在的栈帧信息。

以下是一个使用traceback模块的示例代码：

import traceback

def print_line_number():
    stack = traceback.extract_stack()
    filename, line_number, _, _ = stack[-2]
    print(f'Line {line_number}')

print_line_number()

上述代码中，我们使用traceback.extract_stack()函数来获取当前代码的栈帧信息。然后，通过索引-2来获取上一个栈帧的文件名和行号。最后，我们使用print()函数来打印行号。

需要注意的是，traceback模块还提供了其他有用的函数和方法，用于处理异常和追踪堆栈信息。

5、使用enumerate函数

lines = ['print("Hello, World!")', 'x = 1', 'print(x)']

for i, line in enumerate(lines, start=1):
    print(f'Line {i}: {line}')   

在上面的代码中，enumerate(lines, start=1)表示从1开始对lines中的元素进行迭代，i表示行号，line表示代码行。我们通过f-string将行号和代码行打印出来。

在Python中，可以使用enumerate函数来打印行号。下面是几种不同的实现方法：

方法1：

lines = ["line1", "line2", "line3"]
for i, line in enumerate(lines, start=1):
    print(f"Line {i}: {line}")
    

方法2：

lines = ["line1", "line2", "line3"]
for i in range(len(lines)):
    print(f"Line {i+1}: {lines[i]}")

方法3：

with open('file.txt') as f:
    for i, line in enumerate(f, start=1):
        print(f"Line {i}: {line.strip()}")