如何用Python在2D array上生成指定长度的随机直线？

你的需求本质是生成八连通离散网格直线，刚好可以用成熟的Bresenham直线算法实现，该算法输出的点天然满足横、纵、对角相邻的要求，且形态严格为直线。

具体实现步骤 #

• 第一步：参数合法性校验
  250×250数组内可生成的最长直线为对角线长度，约为353，若指定长度超出2~353范围可直接返回错误。
• 第二步：随机生成符合长度要求的直线端点
  推荐用更简单的随机端点匹配方案：
  循环生成两个随机坐标(x0,y0)、(x1,y1)，范围均为0≤坐标值≤249，计算两点欧氏距离，若和指定长度的误差≤0.5则停止循环，否则重新生成端点。
  如果需要限制直线为水平/垂直/45度正交类型，可在生成端点时强制满足x0=x1/y0=y1/abs(x0-x1)=abs(y0-y1)的条件即可。
• 第三步：生成网格直线点并赋值
  用Bresenham算法计算两个端点之间的所有网格点，直接把对应位置的数组值从0改为1即可。

Python代码示例 #

import numpy as np
import random
import math

# 八连通Bresenham直线生成算法
def bresenham_line(x0, y0, x1, y1):
    points = []
    dx = abs(x1 - x0)
    dy = abs(y1 - y0)
    step_x = 1 if x0 < x1 else -1
    step_y = 1 if y0 < y1 else -1
    err = dx - dy

    while True:
        points.append((x0, y0))
        if x0 == x1 and y0 == y1:
            break
        err_double = 2 * err
        if err_double > -dy:
            err -= dy
            x0 += step_x
        if err_double < dx:
            err += dx
            y0 += step_y
    return points

# 主调用函数
def generate_line_in_array(target_length, arr_size=250):
    # 初始化全零数组
    res_arr = np.zeros((arr_size, arr_size), dtype=np.uint8)
    max_available_len = int(math.hypot(arr_size, arr_size)) + 1
    if target_length < 2 or target_length > max_available_len:
        raise ValueError(f"长度需在2~{max_available_len}范围内")
    
    # 匹配符合长度要求的端点
    while True:
        x0, y0 = random.randint(0, arr_size-1), random.randint(0, arr_size-1)
        x1, y1 = random.randint(0, arr_size-1), random.randint(0, arr_size-1)
        current_len = math.hypot(x1 - x0, y1 - y0)
        if abs(current_len - (target_length - 1)) <= 0.5:
            break
    
    # 赋值直线点
    line_points = bresenham_line(x0, y0, x1, y1)
    # 严格匹配长度，截断多出来的1个点误差
    for x, y in line_points[:target_length]:
        res_arr[y][x] = 1
    return res_arr

内容的提问来源于stack exchange，提问作者Koza Kurumlu