如何用Python编写函数，基于输入字符串生成指定尺寸的矩形图案？

核心规律梳理 #

先观察两个调用示例的输出，不难总结出统一的生成规则：

• 输出为 n 行 n 列的字符矩形，n 是函数的第二个入参
• 输入的字符串会被当成无限循环的字符池，所有要输出的字符按顺序从这个循环字符池中取，取到字符串末尾就自动回到开头重新取
• 第一行取前 n 个字符，第二行紧跟第一行的末尾往后取 n 个字符，第三行再紧跟第二行末尾往后取 n 个，以此类推直到取完 n 行

具体实现步骤 #

你可以按照下面的步骤逐行写代码：

1. 先获取函数的两个入参：用于生成内容的字符串 s，矩形的边长 n
2. 先判断边界情况：如果 s 是空字符串或者 n 小于等于0，可以按作业要求加报错逻辑或者直接返回
3. 循环 n 次生成每一行的内容，循环次数对应第 i 行（从0开始计数）：
   • 第 i 行的第一个字符，对应原字符串 s 的索引为 (i * n) % len(s)（% 是取模运算，用来实现超出长度后回到开头的效果）
   • 遍历当前行的每一个位置 j（从0到n-1），对应 s 的索引为 (i * n + j) % len(s)，把这些索引对应的字符拼接起来就是当前行的内容
4. 逐行打印拼接好的字符串即可

核心逻辑验证（以示例1为例） #

s = "ab"，len(s)=2，n=3

• 第0行：索引分别是 0%2=0、1%2=1、2%2=0，对应字符a、b、a，拼接得aba
• 第1行：索引分别是3%2=1、4%2=0、5%2=1，对应字符b、a、b，拼接得bab
• 第2行：索引分别是6%2=0、7%2=1、8%2=0，对应字符a、b、a，拼接得aba
  和示例输出完全一致。

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