使用自定义函数实现：主函数输入你和同桌的学号；自定义函数交换两个变量的值；主函数输出交换后的学号。

好的，这是一个简单的示例代码：

def swap(a, b):
    temp = a
    a = b
    b = temp
    return a, b

if __name__ == '__main__':
    my_id = input('请输入你的学号：')
    partner_id = input('请输入同桌的学号：')
    my_id, partner_id = swap(my_id, partner_id)
    print('交换后的学号：你的学号是 {}，同桌的学号是 {}'.format(my_id, partner_id))

在这个代码中，我们定义了一个名为 `swap` 的自定义函数，用于交换两个变量的值。在主函数中，我们首先获取你和同桌的学号，然后调用 `swap` 函数将两个学号交换，最后输出交换后的结果。

注意到我们在 `swap` 函数中使用了一个临时变量 `temp` 来保存 `a` 的值，然后将 `a` 赋值为 `b`，再将 `b` 赋值为 `temp`，这样就完成了 `a` 和 `b` 的值交换。最后，我们使用 `return` 语句将交换后的 `a` 和 `b` 返回给主函数。